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() 设置为推理模型就可以了
相关推荐
如竟没有火炬4 分钟前
四数相加贰——哈希表
数据结构·python·算法·leetcode·散列表
码农很忙4 分钟前
从0到1搭建智能分析OBS埋点数据的AI Agent:实战指南
数据库·人工智能
JoannaJuanCV8 分钟前
自动驾驶—CARLA仿真(5)Actors与Blueprints
人工智能·机器学习·自动驾驶
Saniffer_SH8 分钟前
【每日一题】PCIe答疑 - 接大量 GPU 时主板不认设备或无法启动和MMIO的可能关系?
运维·服务器·网络·人工智能·驱动开发·fpga开发·硬件工程
V1ncent Chen16 分钟前
机器是如何识别图片的?:卷积神经网络
人工智能·神经网络·cnn
背心2块钱包邮17 分钟前
第9节——部分分式积分(Partial Fraction Decomposition)
人工智能·python·算法·机器学习·matplotlib
木盏19 分钟前
三维高斯的分裂
开发语言·python
a程序小傲26 分钟前
京东Java面试被问:ZGC的染色指针如何实现?内存屏障如何处理?
java·后端·python·面试
辛勤的程序猿26 分钟前
改进的mamba核心块—Hybrid SS2D Block(适用于视觉)
人工智能·深度学习·yolo