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() 设置为推理模型就可以了
相关推荐
努力犯错1 分钟前
Qwen Image Layered:革命性的AI图像生成与图层分解技术
人工智能·深度学习·计算机视觉
杜子不疼.4 分钟前
【AI】基于GLM-4_7与数字人SDK的政务大厅智能指引系统实践
人工智能·microsoft·政务
core51218 分钟前
SGD 算法详解:蒙眼下山的寻宝者
人工智能·算法·矩阵分解·sgd·目标函数
阿湯哥18 分钟前
Spring AI Alibaba 实现 Workflow 全指南
java·人工智能·spring
Tezign_space19 分钟前
Agent Skills 详解:5大核心能力架构与AI Agent落地实践
人工智能·架构·生成式ai·ai agent·上下文工程·skills·agent skills
m0_4665252921 分钟前
东软添翼AI 2.0获评医疗健康标杆AI Agent TOP10
大数据·人工智能
sky172026 分钟前
VectorStoreRetriever 三种搜索类型
python·langchain
用户51914958484533 分钟前
Linux PAM环境变量注入漏洞利用工具解析
人工智能·aigc
哔哔龙34 分钟前
Langchain中“logprobs”的作用
人工智能
智谱开放平台35 分钟前
理解 Claude 的 Agentic 生态:把零散能力组织成可持续的工作流
人工智能·claude