res_layers in NN.py #2
Description
Hi,
I think the self.res_layers should be rewritten as self.res_layers = nn.Sequential(*[ResLayer()] * res_layer_number),
otherwise Pytorch cannot resist these layers
{{ message }}
Hi,
I think the self.res_layers should be rewritten as self.res_layers = nn.Sequential(*[ResLayer()] * res_layer_number),
otherwise Pytorch cannot resist these layers