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() 设置为推理模型就可以了
相关推荐
一个帅气昵称啊6 分钟前
Net AI智能体开源框架NetCoreKevin为企业AI智能体系统Saas信息化建设赋能-开启智能应用的无限可能
人工智能·开源
yzx99101312 分钟前
卷积神经网络(CNN):深度学习的视觉革命者
人工智能·机器学习
路边草随风22 分钟前
python 调用 spring ai sse mcp
人工智能·python·spring
深圳市快瞳科技有限公司42 分钟前
宠物识别算法在AI摄像头的应用实践:从多宠识别到行为分析
人工智能·智能硬件·宠物
ziwu42 分钟前
【鱼类识别系统】Python+TensorFlow+Django+人工智能+深度学习+卷积神经网络算法
人工智能·深度学习·图像识别
小马爱打代码1 小时前
Spring AI:ChatMemory 实现聊天记忆功能
java·人工智能·spring
ziwu1 小时前
【植物识别系统】Python+TensorFlow+Django+人工智能+深度学习+卷积神经网络算法
人工智能·深度学习·图像识别
Al leng1 小时前
机器学习中偏差和方差的通俗理解
人工智能·机器学习
Dr.Kun1 小时前
【鲲码园Python】基于pytorch的鸟品种分类系统(25类)
pytorch·python·分类
Mxsoft6191 小时前
某次数据解析失败,发现IEC61850版本差异,手动校验报文结构救急!
人工智能