pytorch报错:module 'torch.nn' has no attribute 'init'

在调用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当前路径的规则有些改变。

    原文作者:深度学习入门
    原文地址: https://www.jianshu.com/p/0b23014971a6
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞