Pytorch学习笔记（二）使用Pytorch的常见错误汇总

最新推荐文章于 2024-10-24 15:00:44 发布

cg101202

最新推荐文章于 2024-10-24 15:00:44 发布

阅读量7.1w

点赞数 29

分类专栏： PyTorch 文章标签： pytorch 0.4迁移指南

本文链接：https://blog.csdn.net/qq_38156052/article/details/81293412

版权

那些年我们一起踩过的坑!

文章目录

那些年我们一起踩过的坑!
Error
Warning

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