Hi @Sara_Wasl,
Does it always happen at the same iteration? Maybe the last iteration of the first epoch?
If your dataloader 15 has drop_last  set to False (which is the default value), and if your dataset is not a multiple/aligned on your batch size, the last iteration an epoch will be equal to epoch_size % batch_size.