mxnet fine-tune
网络finetune时，有两种思路，微调整个网络或者仅仅微调网络后几层，现以finetune mobilefacenet为例，加载预训练模型，并固定参数，仅训练自己添加的全连接层
通过查找发现fixed_param_names可达到fix的效果，代码如下
symbol, arg_params, aux_params = mx.model.load_checkpoint('model-...
原创
2018-08-20 20:25:09 ·
1258 阅读 ·
0 评论
关注
分享
扫一扫