bsridatta:

@Usama_Hasan I dont think it has anything to do with nn.Module() but how the dataset is initialized


Hy @bsridatta. You’re right Man, I made the comment considering the heading of the issue.