【debug锦集】Expected input batch_size (1) to match target batch_size (0)
手动调整下label即可。
·
我是雪天鱼，一名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即可
有“AI”的1024 = 2048，欢迎大家加入2048 AI社区
更多推荐
2
0- 0
已为社区贡献21条内容
所有评论(0)