Table Of Contents
Table Of Contents

module_checkpoint

mxnet.callback.module_checkpoint(mod, prefix, period=1, save_optimizer_states=False)[source]

Callback to checkpoint Module to prefix every epoch.

Parameters:
  • mod (subclass of BaseModule) – The module to checkpoint.
  • prefix (str) – The file prefix for this checkpoint.
  • period (int) – How many epochs to wait before checkpointing. Defaults to 1.
  • save_optimizer_states (bool) – Indicates whether or not to save optimizer states for continued training.
Returns:

callback – The callback function that can be passed as iter_end_callback to fit.

Return type:

function