Nefelibat 2021-07-20 10:36 采纳率: 25%
浏览 43

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'

  • 写回答

1条回答 默认 最新

  • 我是小白呀 Python领域优质创作者 2022-11-20 13:42
    关注

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

    评论

报告相同问题?

问题事件

  • 创建了问题 7月20日

悬赏问题

  • ¥15 求差集那个函数有问题,有无佬可以解决
  • ¥15 【提问】基于Invest的水源涵养
  • ¥20 微信网友居然可以通过vx号找到我绑的手机号
  • ¥15 寻一个支付宝扫码远程授权登录的软件助手app
  • ¥15 解riccati方程组
  • ¥15 display:none;样式在嵌套结构中的已设置了display样式的元素上不起作用?
  • ¥15 使用rabbitMQ 消息队列作为url源进行多线程爬取时,总有几个url没有处理的问题。
  • ¥15 Ubuntu在安装序列比对软件STAR时出现报错如何解决
  • ¥50 树莓派安卓APK系统签名
  • ¥65 汇编语言除法溢出问题