I guess the error is raised in this line of code:
features = features.reshape(1, sequence_length, input_size)

Here you are trying to reshape features to a fixed size which doesn’t match the number of elements the tensor contains.
Since you are using batch_size=2 in the DataLoader, I would think you might want to reshape the tensor to keep the batch size equal.
To do so, this could work:
features = features.reshape(features.size(0), -1, input_size)

This operation now keeps the size of dim0 constant, uses input_size for dim2, and flattens the additional values into the temporal dimension dim1.
You should check if the reshape operation is applied correctly, as I don’t know what the original shape of the features tensor was.