utils::save_model Namespace Reference

Functions
def	save_checkpoint

---

Function Documentation

def utils.save_model.save_checkpoint	(		prefix,
			epoch,
			arg_params,
			aux_params
	)

Checkpoint the model data into file.
:param prefix: Prefix of model name.
:param epoch: The epoch number of the model.
:param arg_params: dict of str to NDArray
    Model parameter, dict of name to NDArray of net's weights.
:param aux_params: dict of str to NDArray
    Model parameter, dict of name to NDArray of net's auxiliary states.
:return: None
prefix-epoch.params will be saved for parameters.

Definition at line 4 of file save_model.py.