删除分类网络预训练权重的的head部分的权重以及修改权重名称

于 2022-03-17 10:22:34 发布
阅读量800 收藏 2
点赞数
文章标签： 深度学习 机器学习 人工智能
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.youkuaiyun.com/m0_59135992/article/details/123475116
版权

 创建一个新字典，先将pretrained_dict中的键值遍历一遍，按.分割，如果权重名称中的第一个单词不是head，就放入到新字典中，同时可以进行重命名。

评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值