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() 设置为推理模型就可以了
相关推荐
gCode Teacher 格码致知1 小时前
Python基础教学:Python的openpyxl和python-docx模块结合Excel和Word模板进行数据写入-由Deepseek产生
python·excel
Danceful_YJ1 小时前
33.Transformer架构
人工智能·pytorch·深度学习
Destiny_where3 小时前
Agent平台-RAGFlow(2)-源码安装
python·ai
美狐美颜SDK开放平台3 小时前
美颜SDK性能优化实战:GPU加速与AI人脸美型的融合开发
人工智能·音视频
molunnnn4 小时前
第四章 Agent的几种经典范式
开发语言·python
AI浩4 小时前
VSSD:具有非因果状态空间对偶性的视觉Mamba模型
人工智能·目标检测·计算机视觉
lqqjuly5 小时前
Lidar调试记录Ⅳ之Ubuntu22.04+ROS2+Livox_SDK2环境下编译Livox ROS Driver 2
人工智能·机器人·自动驾驶
qq_436962185 小时前
数据中台:打破企业数据孤岛,实现全域资产化的关键一步
数据库·人工智能·信息可视化·数据挖掘·数据分析
linuxxx1105 小时前
django测试缓存命令的解读
python·缓存·django
宇若-凉凉5 小时前
BERT 完整教程指南
人工智能·深度学习·bert