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

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

这个错误出现的原因有好多种!!
其中一个比较可能的原因是因为:封装dataloader的时候,最后剩下的不足一个batchsize! 自带的dataloader就会有这个现象。

改正:

batch_size_s = len(targets) #不足一个batch_size直接停止训练
if batch_size_s < BATCH_SIZE:
    break

就把最后一个不足batchsize的丢掉!!

你可能感兴趣的:(踩坑,python,深度学习,pytorch)