Table Of Contents
Table Of Contents

mxnet.model.load_checkpoint

mxnet.model.load_checkpoint(prefix, epoch)[source]

Load model checkpoint from file.

Parameters

  • prefix (str) – Prefix of model name.

  • epoch (int) – Epoch number of model we would like to load.

Returns

  • symbol (Symbol) – The symbol configuration of computation network.

  • 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

  • Symbol will be loaded from prefix-symbol.json.

  • Parameters will be loaded from prefix-epoch.params.