Нет параметров в классе Net даже после наследования от nn.Module - PullRequest
0 голосов
/ 10 июля 2020

По какой-то причине мой класс 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()
[]

1 Ответ

0 голосов
/ 10 июля 2020

В коде

def ___init__(self):

вы используете три символа подчеркивания перед init, используйте два: __init__.

...