RuntimeError: The size of tensor a (16) must match the size of tensor b (9) at non-singleton

原因：

BATCHSIZE=16

有一个数据集大小为697

697➗16=43……9

 

解决方案：
设置Dataloader的drop_last为True，drop_last决定不足一个batch_size的数据是否保留。