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