Solved by smth in post #4 
              
              
                you can add drop_last=True to your DataLoader constructor, that should fix the problem. 
Look at documentation on what that option does.