Thanks for your reply @spanev,
Yes that’s true, happened in the last iteration of the first epoch.
No, I didn’t add drop_last
The last iteration number =1667, number of epochs = 50 and the batch_size = 36, do you mean I should chose appropriate batch_size? if yes, could you please tell me how?