Zzn0x0YeYang

I have met the same problem and later I solve with it. You only need to add "drop_last=True" to the data_loader. I hope it will be helpful. Thank u.