Quando eu faço o forward pelo nn.Module estou encontrando o seguinte erro:
class MinhaRede(nn.Module):
#definir arquitetura
def __init__(self, input_size, hidden_size, output_size):
super(MinhaRede, self).__init__()
self.hidden = nn.Linear(input_size, hidden_size)
self.relu = nn.ReLU()
self.output = nn.Linear(hidden_size, output_size)
#gerar uma saída a partir de X
def foward(self, X):
hidden = self.relu(self.hidden(X))
output = self.output(hidden)
return output
input_size = 2
hidden_size = 8
output_size = 1
net = MinhaRede(input_size, hidden_size, output_size)
print(net)
print(X1.shape)
tensor = torch.from_numpy(X1).float()
pred = net(tensor)
print(pred.size())