深度学习DeBug小笔记(二)——_pickle.UnpicklingError: unpickling stack underflow
前言
没啥说的,今天debug了一天,真是美丽祥和的一天呢…
参考于:linux载入pytorch的预训练模型时遇到_pickle.UnpicklingError: unpickling stack underflow
Bug
当时我由于在训练一个骨干网络为resnet101的语义分割网络,当时想要下载预训练模型resnet101时,遇到了以下报错,这个解决方案也适合与其他下载骨干网络未完成,导致的此问题
Traceback (most recent call last):
...
magic_number = pickle_module.load(f)
_pickle.UnpicklingError: unpickling stack underflow
问题原因
这个问题是由于先前已经缓存了resnet101模型的一部分,但是没有缓存完毕,导致临时文件夹中存在部分不完整的模型,载入失败。解决方法是删除原来临时文件夹中的resnet101模型。
解决方法
liunx
liunx下,resnet101未下载完的预训练模型一般在以下两种位置
/home/user/.cache/torch/checkpoints
或者
/home/user/.torch/models
路径中的user
是你的用户名,请按照你的用户名进行更改。
不同系统的具体路径不同,可以两个都尝试一下。直接用ls命令可能无法查看到~/.cache
或者~/.torch
这类隐藏文件夹,直接cd进入目录即可。
删除下载了一半的模型,并且记得定义新的临时文件夹,再次运行代码,解决问题。
windows
windows下一般位于,这个文件夹名下,bambosir920
为你的用户名,请按照你的用户名进行修改
C:\Users\bamboosir920
这里应该是位于这个文件夹下(PS:我当时没记录下来,有些记不清了,2333)
C:\Users\bamboosir920\.ipynb_checkpoints
如果不是的话,在C:\Users\bamboosir920
目录下,搜索后缀为.pth
的文件即可,可以直接搜索到相应的预训练模型,删除即可