Hi. Thanks for your reply! I encountered the same problem and setting drop_last to True really worked. Could you please explain why this change fixes this error? I always set drop_last=False but this is the first time that this error occurred