pytorch调试过程中问题总结(持续更新)

最新推荐文章于 2022-04-12 17:45:33 发布

canfly_zhou

最新推荐文章于 2022-04-12 17:45:33 发布

阅读量807

点赞数

分类专栏： pytorch 文章标签： deep learning

本文链接：https://blog.csdn.net/zhouhuaidong/article/details/90489827

版权

1 篇文章 0 订阅

订阅专栏

1 pkl文件读取问题
问题代码:

with open('data.pkl', 'rb') as fp:
   	 sample_datas = pickle.load(fp)

错误信息:
UnicodeDecodeError: ‘ascii’ codec can’t decode byte 0xc7 in position 1: ordinal not in range(128)
解决方法:

with open('data.pkl', 'rb') as fp:
    sample_datas = pickle.load(fp, encoding='iso-8859-1')

2 CUDNN在模型训练过程中报错
问题描述:
有时候在pytorch训练模型过程中CUDNN会提示报错.例如:

net = ResNet101()
net = net.cuda()
cudnn.benchmark = True
net.train()

错误信息:
RuntimeError: cuDNN error: CUDNN_STATUS_EXECUTION_FAILED
解决方法:
此种方法可在网络后面添加torch.backends.cudnn.enabled=False,如:

net = ResNet101()
net = net.cuda()
cudnn.benchmark = True
torch.backends.cudnn.enabled=False
net.train()

自己在学习中记录总结问题,也方便以后遇见类似问题,快速找到解决方法,遇见新的问题逐步更新~~~

关注