首先,该方法适用前提是你的pytorch已经安装成功啦。
此时,在PyCharm的console里是可以引用成功的,但在文件中却导入报错。
原因
transformer的信息没有写入modules的_init_.pyi提示文件中,最后加上就可以了。
解决方法
可以从init文件复制,需要用as命名一下。
我是在anaconda的py37环境里安装的torch, 所以进入顺序为
D:\Anaconda3\envs–> py37–> lib–>site-packages–>torch–>nn–>modules–>init.pyi
这个只要找到你安装Pytorch的包就行了,python也适用。
打开__init__.pyi文件,发现里面是没有transformer这个模块的导入的。
从__init__.py复制
到__init__.pyi,改写成
就行了。