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

问题
python 复制代码
        self.sgf_net.requires_grad_(False)

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

python 复制代码
        # self.sgf_net.requires_grad_(False)
        for param in self.sgf_net.parameters():
            param.requires_grad = False
        self.sgf_net.eval()
```
添加 model.eval() 设置为推理模型就可以了
相关推荐
小二·7 分钟前
Python Web 开发进阶实战:安全加固实战 —— 基于 OWASP Top 10 的全栈防御体系
前端·python·安全
唐叔在学习10 分钟前
还在申请云服务器来传输数据嘛?试试P2P直连吧
后端·python
zhengfei61111 分钟前
【开源渗透工具】——一个开源的多模态大型语言模型红队框架OpenRT
人工智能·语言模型·开源
黄宝康11 分钟前
sublimetext 运行python程序
开发语言·python
WJSKad123513 分钟前
工业零件识别与分类:基于lad_r50-paa-r101_fpn_2xb8_coco_1x模型实现
人工智能·分类·数据挖掘
千汇数据的老司机22 分钟前
靠资源拿项目VS靠技术拿项目,二者的深刻区分。
大数据·人工智能·谈单
聚城云-GeecityCloud39 分钟前
物业行业:在矛盾与转型中回归服务本质
人工智能·数据挖掘·回归
matlabgoodboy1 小时前
程序代做python代编程matlab定制代码编写C++代写plc设计java帮做
c++·python·matlab
a3158238061 小时前
基于大语言模型的新闻判断技术
人工智能·语言模型·自然语言处理
亚里随笔1 小时前
超越LoRA:参数高效强化学习方法的全面评估与突破
人工智能·深度学习·机器学习·lora·rl