代码运行时错误解决
在实战过程中,将遇到的各式各样的错误进行总结,并找到其解决方式!!!
大梦冲冲冲
这个作者很懒,什么都没留下…
展开
-
RuntimeError: Attempting to deserialize object on a CUDA device but torch.cuda.is_available() is Fal
错误描述:这是由于在数据加载的过程中使用了GPU修改方式:修改前位置:torch.load(model)修改后结果:torch.load(model, map_location='cpu')我的代码中修改后的样子:原创 2022-01-21 17:01:13 · 1241 阅读 · 0 评论 -
ValueError:Expected more than 1 value per channel when training,got input size torch.size([])
首先,遇到的错误是:ValueError:Expected more than 1 value per channel when training…解决方式:1、复制问题,百度。在博客上找到了很多的相关处理方式,大部分都是认为这是一个训练和测试时通道不一致的问题。但是,我仔细检查了我的网络构造,我觉得在这个方面好像没啥问题。2、带着疑问,继续百度。最后,终于发现了问题所在——“在网络训练过程中,我们的batchsize(例如3)的设置,会将数据集(例如10)平均分开。但如若数据集的总数量,不能整.原创 2021-12-24 15:37:05 · 1837 阅读 · 0 评论