def re_initializer_layer(model, num_classes, layer=None):
"""remove the last layer and add a new one"""
indim = nn.Linear.in_features
private_key = model.module.linear
if layer:
model.module.linear = layer
else:
model.module.linear = nn.Linear(indim, num_classes).cuda()
return model, private_key
这是我的代码,是为了移除网络的最后一层,添加新的一层,结果报错
AttributeError: type object 'Linear' has no attribute 'in_features'