AttributeError: module ‘torchvision.models‘ has no attribute ‘xxxx‘ 问题和解决

学校给小虎配的电脑比较low，没有GPU，所以安装的是cpu版本的pytorch。这就导致版本比较低，所以torchvision的一些模型用不了。另外我在kaggle试的时候，发现也没有EfficientNet。

具体问题

我是试用efficientnet出现的情况，报错如下：

AttributeError: module 'torchvision.models' has no attribute 'efficientnet_b0'

解决方案

把源码复制过来跑。
比如EfficientNet的源码在网址：
SOURCE CODE FOR TORCHVISION.MODELS.EFFICIENTNET
不过里面的import的依赖可能路径也要改一改，或者自己把源码复制到模型的.py文件里面。

参考资料

大部分经典模型可以在这里找到源码：
TORCHVISION.MODELS

另外，重装更新pytorch和torchvision小虎试过，暂时不起作用。