May be my question is naive and it has an answer in the python documentation. But so
I am trting to build such a network structure
N = 5
INPUT_SIZE = 10
class Net(nn.Module):
def __init__(self, input_size):
super(Net, self).__init__()
self.input_size = input_size
self.in_out_size = [input_size + N - i for i in range(input_size)]
self.line = [nn.Linear(self.in_out_size[i], self.in_out_size[i+1]) for i in range(input_size-1)]
def forward(self, x):
for i in range(self.input_size):
x = self.line[i](x)
return x
But I can’t do it because I have empty the parameters list.
if __name__ == '__main__':
model = Net(INPUT_SIZE)
print(model.parameters())
[]
Can I use a cycles of kind self.line = [nn.Linear/nn.GRU/nn.CNN for i in range()]
for bilding of network?