custom fc net pytorch

class NetLong(nn.Module):
    def __init__(self):
        super(NetLong, self).__init__()
        self.fcs = self.make_layers()        
        
    def forward(self, x):        
        x = self.fcs(x)        
        return x
    
    def make_layers(self):
        layers = []
        _in_layer = 512
        diff = 128
        for _out_layer in range(_in_layer-diff, 2, -diff):
            layers += [nn.Linear(_in_layer, _out_layer)]
            layers += [nn.ReLU()]
             layers += [nn.BatchNorm1d(_out_layer)]
            layers += [nn.Dropout(0.5)]
            _in_layer = _out_layer
        layers += [nn.Linear(_in_layer, 2)]
        return nn.Sequential(*layers)