mxnet Module load

load函数：

其实就是

sym, arg_params, aux_params = mx.model.load_checkpoint(r"mv2_55\model", 0)
 

load源码： 

  

   

    

     

    
    

     

       @staticmethod
     
    
   

    

     

    
    

     
 def load(prefix, epoch, load_optimizer_states=False, **kwargs):
     
    
   

    

     

    
    

     
 """Creates a model from previously saved checkpoint.
     
    
   

    

     

    
    

     

      
     
    
   

    

     

    
    

     

       Parameters
     
    
   

    

     

    
    

     

       ----------
     
    
   

    

     

    
    

     

       prefix : str
     
    
   

    

     

    
    

     

       path prefix of saved model files. You should have
     
    
   

    

     

    
    

     

       "prefix-symbol.json", "prefix-xxxx.params", and
     
    
   

    

     

    
    

     

       optionally "prefix-xxxx.states", where xxxx is the
     
    
   

    

     

    
    

     

       epoch number.
     
    
   

    

     

    
    

     

       epoch : int
     
    
   

    

     

    
    

     

       epoch to load.
     
    
   

    

     

    
    

     

       load_optimizer_states : bool
     
    
   

    

     

    
    

     

       whether to load optimizer states. Checkpoint needs
     
    
   

    

     

    
    

     

       to have been made with save_optimizer_states=True.
     
    
   

    

     

    
    

     

       data_names : list of str
     
    
   

    

     

    
    

     

       Default is `('data')` for a typical model used in image classification.
     
    
   

    

     

    
    

     

       label_names : list of str
     
    
   

    

     

    
    

     

       Default is `('softmax_label')` for a typical model used in image
     
    
   

    

     

    
    

     

       classification.
     
    
   

    

     

    
    

     

       logger : Logger
     
    
   

    

     

    
    

     

       Default is `logging`.
     
    
   

    

     

    
    

     

       context : Context or list of Context
     
    
   

    

     

    
    

     

       Default is ``cpu()``.
     
    
   

    

     

    
    

     

       work_load_list : list of number
     
    
   

    

     

    
    

     

       Default ``None``, indicating uniform workload.
     
    
   

    

     

    
    

     

       fixed_param_names: list of str
     
    
   

    

     

    
    

     

       Default ``None``, indicating no network parameters are fixed.
     
    
   

    

     

    
    

     

       """
     
    
   

    

     

    
    

     

       sym, args, auxs = load_checkpoint(prefix, epoch)
     
    
   

    

     

    
    

     

       mod = Module(symbol=sym, **kwargs)
     
    
   

    

     

    
    

     

       mod._arg_params = args
     
    
   

    

     

    
    

     

       mod._aux_params = auxs
     
    
   

    

     

    
    

     

       mod.params_initialized = True
     
    
   

    

     

    
    

     
 if load_optimizer_states:
     
    
   

    

     

    
    

     

       mod._preload_opt_states = '%s-%04d.states'%(prefix, epoch)
     
    
   

    

     

    
    

     
 return mod
     
    
  
 

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

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