在調用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當前路徑的規則有些改變。