1.6.0GPU pycharm2020.2
在jupyter中nn是有提示的,但是pycharm里没有。找了半天只有很久很久以前的解决方法
受一篇博客的启发,在加载时这么写,暂时性的解决了
import torch.nn.modules as nn
或
import torch.nn.modules as nn
from torch.nn import init
modules里的用nn.****
其他如init的直接init.normal_
不知道是我自己的问题还是pycharm与torch本身的问题