在调用pytorch的init模块的时候,出现错误:
module ‘torch.nn’ has no attribute ‘init’
但是在github中看到是有这个文件的,所以再看了下本地安装路径,也有这个文件。想到应该是没有import的原因,就把文件~/anaconda3/lib/python3.6/site-packages/torch/nn/init.py中的代码增加一行:
from .modules import *
from .parameter import Parameter
from .parallel import DataParallel
from .init import *
问题解决。
ps:这在python2中是没有这个问题的,python3对于import当前路径的规则有些改变。