AttributeError: module 'torch.ao.nn' has no attribute 'Module' 这个问题如何解决呀 大佬们
应该是torch.nn把torch.ao是什么
不知道你这个问题是否已经解决, 如果还没有解决的话:最近在学pytorch,然后想手敲一下pytorch对于LeNet的实现已加深理解.然后就有了一下的笑话:
conda create -n pytorch0.4.1 python=3
;class Net(nn.Module):
,换了一个有的参数class Net(nn.ScriptModule):
;import torch as nn
,而官方用的是import torch.nn as nn