pytorch代码报错

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'

报错的原因: 在没有重置模型权重的情况下更改任务. 你看一下修改前网络是几分类.