indim = input_shp[0]
self.linear0 = nn.Linear(indim, config.num_unit)
for i in range(1, config.num_hidden-1):
    setattr(self, "linear"+str(i), nn.Linear(config.num_unit, config.num_unit))
setattr(self, "linear"+str(config.num_hidden-1), nn.Linear(config.num_unit, config.num_class))

self.output = nn.Linear(indim, config.num_class)