如果只更新某个 dense 层的权重：Trainer 里面用下面这个而不是所有的 model.collect_params()
model.collect_params(select='.*dense0')
如果想设置为只有 dense0 层不更新
model.collect_params('.*dense0').setattr('grad_req','null')
如果只更新某个 dense 层的权重：Trainer 里面用下面这个而不是所有的 model.collect_params()
model.collect_params(select='.*dense0')
如果想设置为只有 dense0 层不更新
model.collect_params('.*dense0').setattr('grad_req','null')
本文标题：MXNet 冻结网络层
本文链接：https://www.haomeiwen.com/subject/yeouzktx.html
网友评论