发现在创建模型时出现了这个问题,出现的原因是我的模型初始化需要加载一个预训练的resnet50模型,但是resnet50模型完整的有97.7MB,而我只下载了96.3MB就断了,导致初始化的时候无法加载完整模型报错。
解决办法是手动下载resnet50的预训练模型,并且把该预训练的模型放到C:\Users\86158.cache\torch\hub\checkpoints这一目录下。
unpickling stack underflow
最新推荐文章于 2023-09-16 18:53:29 发布