您好,欢迎访问这里是您的网站名称官网!
新闻资讯

服务热线400-123-4567

常见问答

首页 > 新闻资讯 > 常见问答

用自己训练的权重训练其他模型出现的问题ValueError: loaded state dict contains a parameter group that doesn‘t match the s

作者:佚名 发布时间:2024-04-15 13:00:05点击:
所以,在读取优化器记录的时候要注意,如果保存的数据是针对某些层保存的,而加载的时候又希望对所有层进行训练,这时候优化器数据就没必要加载了,要么写个try,要么就把这行注释掉。在模型中如果设置了requires_grad=True,则表示该层要进行梯度计算,标记为False则不计算梯度,在迁移学习中一般会设置成False,这样会大量减少算力。将模型编辑成我们需要的样子,例如将全连接层的输出定义成我们要的输出。这样做的好处是可以节省计算梯度和优化的算力。我在代码中会保存优化器以便于之后的继续测试。
相关标签: 优化 梯度

平台注册入口