Sorry my bad, I thought correcting shape of dataloader solved the problem but the error came back.
Also I don’t understand the reason, my __getitem__ function returns tuple of numpy arrays, I dont understand why would error about FloatTensor and DoubleTensor would come up