由于某些原因,我的class-Net没有参数,即使我已经正确导入了所有内容。,它仍然显示net没有参数或属性,即使在打印net时,它也只打印空列表。有人能告诉我怎么了吗?
import torch.nn as nn
import torch.nn.functional as F
class Net(nn.Module):
def ___init__(self):
super(Net, self).__init__()
self.fc1 = nn.Linear(784, 64)
self.fc2 = nn.Parameter(nn.Linear(64, 64))
self.fc3 = nn.Parameter(nn.Linear(64, 64))
self.fc4 = nn.Parameter(nn.Linear(64, 10))
def forward(self, x):
x = F.relu(self.fc1(x))
x = F.relu(self.fc2(x))
x = F.relu(self.fc3(x))
x = self.fc4(x)
return F.log_softmax(x, dim=1)
net = Net()
print(net)
print(list(net.parameters()))
输出:
Net()
[]
在代码中
def ___init__(self):
如果在init
之前使用三个下划线,请使用两个:__init__
。