Table Of Contents

mxnet.model.save_checkpoint

mxnet.model.save_checkpoint(prefix, epoch, symbol, arg_params, aux_params)

Checkpoint the model data into file.

Parameters

• prefix (str) – Prefix of model name.

• epoch (int) – The epoch number of the model.

• symbol (Symbol) – The input Symbol.

• 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.

Notes

• prefix-symbol.json will be saved for symbol.

• prefix-epoch.params will be saved for parameters.