Hello,
Apologies for the late reply. The dataset itself is not to blame per se, however, an issue might be the scale of the data. Could you please print out a random sample from the dataset, as well as its range (e.g., print(data.min(), data.max()))? I ask because it’s possible that the data falls in the range [0, 255], whereas PyTorch expects it to be normalized by a factor of 255.
Thank you.