Pytorch如何根据layers的name实现冻结训练方式-创新互联

这篇文章主要介绍了Pytorch如何根据layers的name实现冻结训练方式，具有一定借鉴价值，感兴趣的朋友可以参考下，希望大家阅读完这篇文章之后大有收获，下面让小编带着大家一起了解一下。

创新互联专注于新乡网站建设服务及定制，我们拥有丰富的企业做网站经验。 热诚为您提供新乡营销型网站建设，新乡网站制作、新乡网页设计、新乡网站官网定制、微信小程序开发服务，打造新乡网络公司原创品牌,更为您提供新乡网站排名全网营销落地服务。

使用model.named_parameters()可以轻松搞定，

model.cuda()
 
 
# ######################################## Froze some layers to fine-turn the model ########################
for name, param in model.named_parameters(): # 带有参数名的模型的各个层包含的参数遍历
  if 'out' or 'merge' or 'before_regress' in name: # 判断参数名字符串中是否包含某些关键字
    continue
  param.requires_grad = False
# #############################################################################################################
 
 
optimizer = optim.SGD(filter(lambda p: p.requires_grad, model.parameters()),
           lr=opt.learning_rate * args.world_size, momentum=0.9, weight_decay=5e-4)

感谢你能够认真阅读完这篇文章，希望小编分享的“Pytorch如何根据layers的name实现冻结训练方式”这篇文章对大家有帮助，同时也希望大家多多支持创新互联，关注创新互联行业资讯频道，更多相关知识等着你来学习!

网站栏目：Pytorch如何根据layers的name实现冻结训练方式-创新互联
本文地址：https://www.cdcxhl.com/article26/dscccg.html

成都网站建设公司_创新互联，为您提供网站制作、服务器托管、面包屑导航、ChatGPT、搜索引擎优化、手机网站建设

广告

声明：本网站发布的内容（图片、视频和文字）以用户投稿、用户转载内容为主，如果涉及侵权请尽快告知，我们将会在第一时间删除。文章观点不代表本网站立场，如需处理请联系客服。电话：028-86922220；邮箱：631063699@qq.com。内容未经允许不得转载，或转载时需注明来源： 创新互联