mxnet迁移学习 注释版

如果设置这个参数：

mod_new.binded=True

bind()里面有判断，调用过了，就不会再执行bind()内存了，获取参数会为空

       sym, arg_params, aux_params = mx.model.load_checkpoint(r"E:\jinji\AidLearning-FrameWork-master\src\facencnn\models\mv2_55\model", 0)
       sym_new=get_symbol(512)
       mod_new=mx.mod.Module(symbol=sym_new,context=[mx.cpu()])#,label_names=None)
      # step1 这里获取参数为空
      # mod_new.params_initialized=True
      # mod_new.get_params()
       data_shape_w=112
      # mod_new.binded=True
      #step2
       mod_new.bind(for_training=False,data_shapes=[('data',(1,3,data_shape_w,data_shape_w))])
      #step2 此时网络会生成参数
       arg_params_new=dict()
       aux_params_new=dict()
      for key in arg_params.keys():
       arg_params_new[key]=arg_params[key]
      for key in aux_params.keys():
       aux_params_new[key]=aux_params[key]
      #step3 这里获取参数，就有了
       mod_new.params_initialized=True
       mod_new.get_params()
       mod_new.set_params(arg_params_new,aux_params_new,allow_missing=True)
       mod_new.save_checkpoint("aaaa",0)
  
 

知识点：

mod_new.save_checkpoint("aaaa",0)

aaaa是模型文件名字，0是epoch，是个数字，也可以自行修改。

文章来源: blog.csdn.net，作者：网奇，版权归原作者所有，如需转载，请联系作者。

原文链接：blog.csdn.net/jacke121/article/details/115767733