Which one is more PyTorch like (better) code for nn.Sequential?
def __init__(self):
self.layers = nn.Sequential (layer1, layer2, ...)
def forward(self, x):
self.layers(x)
or
def __init__(self):
self.layer1, self.layer2, ...
def forward(self, x):
ss = nn.Sequential (layer1, layer2, ...)
Both would be okay?