- 批量定义网络层,使用setattr方法
class AModel(nn.Module):
def __init__(self):
super(AModel, self).__init__():
for i in range(5):
setattr(self, 'linear_feature{}'.format(i), nn.Linear(512, 64))
class AModel(nn.Module):
def __init__(self):
super(AModel, self).__init__():
for i in range(5):
setattr(self, 'linear_feature{}'.format(i), nn.Linear(512, 64))