使用torchvision.models.inception_v3(pretrained=True)加载预训练的模型每次都特别慢
欢迎大家关注笔者,你的关注是我持续更博的最大动力原创文章,转载告知,盗版必究使用torchvision.models.inception_v3(pretrained=True)预训练的模型每次加载特别慢文章目录:1 问题描述2 问题解决1 问题描述在使用torchvision.models.inception_v3(pretrained=True)预训练模型时,第一次执行的时候会把预训练的模型下载
·
1 问题描述
在使用torchvision.models.inception_v3(pretrained=True)
预训练模型时,第一次执行的时候会把预训练的模型下载到用户的家目录下的.cache/torch/checkpoints
目录下,下次再执行的时候就会先到该目录下检索,有没有下载好的预训练模型!
但是我在第二次执行该代码时,就一直在加载等待的过程,特别缓慢,也没有报任何的错误,这问题是由scipy这个包导致的!
2 问题解决
torchvision在加载完模型后需要对模型进行参数初始化
,用到了scipy
中的stats.truncnorm.rvs
方法。如果你是安装的Anaconda,默认的scipy是1.4版本
,而该版本速度特别慢别特别慢
,从而导致torchvision运行时间变长!
解决方法:
把scipy更新到最新版本
pip install scipy==1.5.1


♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠
更多推荐
所有评论(0)