【debug锦集】Expected input batch_size (1) to match target batch_size (0)

我是雪天鱼，一名FPGA技术爱好者，B站账号：https://space.bilibili.com/397002941?spm_id_from=333.1007.0.0

（1）由于输入到 loss函数的label不满足要求格式导致

loss = criterion(output, label)

这里的label是tensor类型，元素个数要与种类个数对应，如四分类，则 label=[1,0,0,0]、[0,1,0,0]、[0,0,1,0]、[0,0,0,1] 这样 ，即与预测的输出是一个样子，size和dtype都要一致：
预测输出：

报错details：

解决方法：

手动调整下label即可