Hi,
Datasets and Dataloaders are generally instantiated outside the training/testing loops. Are you sure you instantiate those within the training loop? To delete: how about del dataset/dataloader?
Cheers!