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
♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠