当前位置: 代码迷 >> 综合 >> RuntimeError: The size of tensor a (32) must match the size of tensor b (28) at non-singleton
  详细解决方案

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

热度:96   发布时间:2023-12-15 00:50:31.0

这个错误主要是因为,数据的数量不能整除batch size的数量,改正方法有两种

1. 将batch size改称可以被 数据量整除即可。

2. 将dataloader 中的 drop_last改称False也可。

推荐使用第二种方法更加简单。 

 

  相关解决方案