I don’t think it should really matter unless your dataloader is missing some functions (which should throw an error). If the data and label tensors returned by your dataset are reasonable it is likely the problem is somewhere else.