mxnet.callback.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