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 分钟前
人工智能之数据分析 Pandas:第六章 数据清洗
人工智能·数据分析·pandas
geneculture6 分钟前
融合全部讨论精华的融智学认知与实践总览图:掌握在复杂世界中锚定自我、有效行动、并参与塑造近未来的元能力
大数据·人工智能·数据挖掘·信息科学·融智学的重要应用·信智序位·全球软件定位系统
闲人编程7 分钟前
GraphQL与REST API对比与实践
后端·python·api·graphql·rest·codecapsule
永霖光电_UVLED11 分钟前
安森美与英诺赛科将合作推进氮化镓(GaN)功率器件的量产应用
人工智能·神经网络·生成对抗网络
Dev7z13 分钟前
基于深度学习的脑肿瘤自动诊断和分析系统的研究与实现(Web界面+数据集+训练代码)
人工智能·深度学习
珠海西格电力16 分钟前
零碳园区数字感知基础架构规划:IoT 设备布点与传输管网衔接设计
大数据·运维·人工智能·物联网·智慧城市·能源
AI即插即用18 分钟前
即插即用系列 | WACV 2024 D-LKA:超越 Transformer?D-LKA Net 如何用可变形大核卷积刷新医学图像分割
图像处理·人工智能·深度学习·目标检测·计算机视觉·视觉检测·transformer
草莓熊Lotso23 分钟前
《算法闯关指南:动态规划算法--斐波拉契数列模型》--04.解码方法
c++·人工智能·算法·动态规划
winfredzhang26 分钟前
深入剖析 wxPython 配置文件编辑器
python·编辑器·wxpython·ini配置
狂放不羁霸26 分钟前
电子科技大学2025年机器学习期末考试回忆
人工智能·机器学习