欢迎来到尧图网

客户服务 关于我们

您的位置:首页 > 教育 > 高考 > 1. pytorch 中冻结模型参数后参数仍会被调整

1. pytorch 中冻结模型参数后参数仍会被调整

2025/2/24 19:28:58 来源:https://blog.csdn.net/dododododoooo/article/details/143595834  浏览:    关键词:1. pytorch 中冻结模型参数后参数仍会被调整
问题
        self.sgf_net.requires_grad_(False)

起初设置 requires_grad(False) 优化器也没有添加sgfnet的模型参数。但是在pylightning框架中,每次推理完模型的参数都会被改变,经过仔细排查发现问题

        # self.sgf_net.requires_grad_(False)for param in self.sgf_net.parameters():param.requires_grad = Falseself.sgf_net.eval()
```
添加 model.eval() 设置为推理模型就可以了

版权声明:

本网仅为发布的内容提供存储空间,不对发表、转载的内容提供任何形式的保证。凡本网注明“来源:XXX网络”的作品,均转载自其它媒体,著作权归作者所有,商业转载请联系作者获得授权,非商业转载请注明出处。

我们尊重并感谢每一位作者,均已注明文章来源和作者。如因作品内容、版权或其它问题,请及时与我们联系,联系邮箱:809451989@qq.com,投稿邮箱:809451989@qq.com

热搜词