utils::load_model Namespace Reference

Functions
def	convert_context
def	load_checkpoint
def	load_param

---

Function Documentation

def utils.load_model.convert_context	(		params,
			ctx
	)

:param params: dict of str to NDArray
:param ctx: the context to convert to
:return: dict of str of NDArray with context ctx

Definition at line 27 of file load_model.py.

def utils.load_model.load_checkpoint	(		prefix,
			epoch
	)

Load model checkpoint from file.
:param prefix: Prefix of model name.
:param epoch: Epoch number of model we would like to load.
:return: (arg_params, aux_params)
arg_params : dict of str to NDArray
    Model parameter, dict of name to NDArray of net's weights.
aux_params : dict of str to NDArray
    Model parameter, dict of name to NDArray of net's auxiliary states.

Definition at line 4 of file load_model.py.

def utils.load_model.load_param	(		prefix,
			epoch,
			convert = False,
			ctx = None,
			process = False
	)

wrapper for load checkpoint
:param prefix: Prefix of model name.
:param epoch: Epoch number of model we would like to load.
:param convert: reference model should be converted to GPU NDArray first
:param ctx: if convert then ctx must be designated.
:param process: model should drop any test
:return: (arg_params, aux_params)

Definition at line 39 of file load_model.py.