pytorch 查看参数是否被训练 require_grad()

遍历named_parameters()中的所有的参数，只打印那些param.requires_grad=True的变量。for name, param in model.named_parmeters():if param.requires_grad:print(name)

Golden-sun

7969人浏览 · 2020-07-11 21:17:28

Golden-sun · 2020-07-11 21:17:28 发布

遍历named_parameters()中的所有的参数，只打印那些param.requires_grad=True的变量。

 for name, param in model.named_parameters():
      if param.requires_grad:
          print(name)

2048 AI社区

有“AI”的1024 = 2048，欢迎大家加入2048 AI社区

更多推荐

[ 开源 ] FastAPI + LangGraph 实战智能客服 Agent：从工单分类到自动回复与业务回写附github

2048 AI社区

大厂Java面试实战：Spring Boot/Cloud、Redis/Kafka、JVM调优与Spring AI RAG（内容社区UGC+AIGC客服场景）

以“内容社区+UGC+AIGC智能客服”为业务背景，模拟大厂Java面试：严肃面试官循序渐进提问Spring Boot/Cloud、JVM、Redis、Kafka、数据库、观测、CI/CD与Spring AI/RAG/Agent等，小Y简单题会答复杂题含糊。文末给出每题详细答案与落地方案，便于小白系统学习。