pytorch模型微调——不训练某些层的参数

最新推荐文章于 2025-07-07 12:02:11 发布

农夫山泉2号

最新推荐文章于 2025-07-07 12:02:11 发布

阅读量3.8k

点赞数 2

CC 4.0 BY-SA版权

分类专栏： pytorch 文章标签： pytorch

本文链接：https://blog.youkuaiyun.com/u011622208/article/details/95616220

pytorch 专栏收录该内容

80 篇文章

订阅专栏

博客围绕PyTorch展开，介绍了通过net的named_parameters()属性进行遍历的相关代码及解释，属于信息技术领域中深度学习框架的应用内容。

code

  ### //不训练某些层
  frozen_layers = [net.cnn, net.rnn, net.layer0, net.layer0_1]
  for layer in frozen_layers:
    for name, value in layer.named_parameters():
      value.requires_grad = False
  params = filter(lambda p: p.requires_grad, net.parameters())