【MindSpore产品】【模型部分参数冻结功能】mindspore如何固定住部分层的可训练参数

问题描述:

mindspore如何固定住部分层的可训练参数? layer.trainable = False, 的等效实现是怎样的? 求大佬解答。 

猜测是set_train,但是文档解释并不清晰,“另一个分支”不太理解是啥?

 

已解决:

示例如下

 

解答:

建议参考下这里https://mindspore.cn/tutorials/zh-CN/r1.7/advanced/network/optim.html?highlight=%E4%BC%98%E5%8C%96%E5%99%A8#%E5%8F%82%E6%95%B0%E9%85%8D%E7%BD%AE实现

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值