@Usama_Hasan I dont think it has anything to do with nn.Module() but how the dataset is initialized
@Archy_dragon  It would be helpful if you could provide the error log. Why is the LFW1 dataset not inheriting the Dataset class?