那些年我们一起踩过的坑!
文章目录
Error
标签范围问题
这个错误出现在我参考别人的源码来训练一个车型识别模型,共196类,设置输出num_classes = 196
,结果报下面错误,原因是我的输入的标签是1-196,超出索引范围,正确标签范围应该是[0,num_classes)
_, pred = output.data.cpu().topk(1, dim=1)
RuntimeError: cuda runtime error (59) : device-side assert triggered at /pytorch/aten/src/THC/generic/THCTensorCopy.c:70
可参考:Issue #1204 · pytorch/pytorch
输入图像大小不一的问题
数据集图像大小不一,加载训练集时进行了Resize and Crop ,但是在测试时忘记了,因此出现了以下报错信息:
for batch_idx, (input, target) in enumerate(loader):
File "/usr/local/lib/python2.7/dist-packages/torch/utils/data/dataloader.py", line 264, in __next__
batch = self.collate_fn([self.dataset[i] for i in indices])
File "/usr/local/lib/python2.7/dist-packages/torch/utils/data/dataloader.py", line 138, in default_collate
return [default_collate(samples) for samples in transposed]
File "/usr/local/lib/python2.7/dist-packages/torch/utils/data/dataloader.py", line 115