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() 设置为推理模型就可以了
相关推荐
superman超哥1 分钟前
仓颉内存分配优化深度解析
c语言·开发语言·c++·python·仓颉
一车小面包3 分钟前
大模型与检索系统集成开发核心知识点总结
python
2401_841495645 分钟前
并行程序设计与实现
c++·python·算法·cuda·mpi·并行计算·openmp
invicinble5 分钟前
java集合类(二)--map
java·开发语言·python
算法与编程之美5 分钟前
不同的优化器对分类精度的影响以及损失函数对分类精度的影响.
人工智能·算法·机器学习·分类·数据挖掘
Black蜡笔小新5 分钟前
户外无电无网视频汇聚平台EasyCVR太阳能4G视频监控解决方案
人工智能
sali-tec6 分钟前
C# 基于halcon的视觉工作流-章71 深度学习-预处理OCR
开发语言·人工智能·深度学习·数码相机·算法·计算机视觉·ocr
代码洲学长6 分钟前
文本数据分析的基础知识
python·自然语言处理·数据分析
xzl049 分钟前
当使用 AutoTokenizer 加载百川(Baichuan)模型时出现 BaiChuanTokenizer 相关报错
人工智能·pytorch·python
yangshuo12819 分钟前
心灵宝石MCP部署完全指南:AI IDE积分零损耗的实现方案
ide·人工智能·microsoft