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() 设置为推理模型就可以了
相关推荐
阿达_优阅达22 分钟前
HubSpot 营销指南 | AI 时代,如何同时做好 SEO 与 AEO?
人工智能·ai·seo·营销自动化·hubspot·aeo·sales
趣知岛29 分钟前
智能家居与物联网项目实战全指南:从架构设计到落地部署
python·物联网·智能家居
龘龍龙1 小时前
Python基础(八)
开发语言·python
kkce1 小时前
vsping 推出海外检测节点的核心目的
大数据·网络·人工智能
bin91531 小时前
当AI优化搜索引擎算法:Go初级开发者的创意突围实战指南
人工智能·算法·搜索引擎·工具·ai工具
人工智能技术咨询.1 小时前
深度学习—卷积神经网络
人工智能
机器之心1 小时前
Manus被收购,智谱也定了8天后上市
人工智能·openai
王中阳Go1 小时前
手把手教你用 Go + Eino 搭建一个企业级 RAG 知识库(含代码与踩坑)
人工智能·后端·go
Coder个人博客2 小时前
Llama.cpp 整体架构分析
人工智能·自动驾驶·llama
vibag2 小时前
FastAPI框架
python·pycharm·fastapi