saealib.CheckpointCallback#

class saealib.CheckpointCallback(path, interval=1, format='npz', delete_on_success=False, optimizer=None)[ソース]#

ベースクラス: object

Callback that saves checkpoints at regular generation intervals.

Two formats are supported:

  • 'npz': portable numpy format (best-effort reproducibility).

  • 'pickle': full optimizer + context state (limited complete reproducibility; version-sensitive).

  • 'both': saves both formats each checkpoint.

Register this callback with register(), or pass it implicitly via the checkpoint_* parameters of run() and iterate().

パラメータ:

  • path (str or Path) -- Directory where checkpoint files are saved. Created if absent.

  • interval (int, optional) -- Save a checkpoint every interval generations. Default: 1.

  • format ({'npz', 'pickle', 'both'}, optional) -- Checkpoint format. Default: 'npz'.

  • delete_on_success (bool, optional) -- If True, all checkpoint files produced by this callback are deleted when the run terminates normally (RunEndEvent). Useful to avoid accumulating files after a successful run. Default: False.

  • optimizer (Optimizer or None, optional) -- Required when format is 'pickle' or 'both'.

Methods

__init__

register

Register generation-end and run-end handlers on cbmanager.

Method Details

CheckpointCallback.__init__(path, interval=1, format='npz', delete_on_success=False, optimizer=None)[ソース]#
パラメータ:

  • path (str | Path)

  • interval (int)

  • format (str)

  • delete_on_success (bool)

  • optimizer (Optimizer | None)

戻り値の型:

None

CheckpointCallback.register(cbmanager)[ソース]#

Register generation-end and run-end handlers on cbmanager.

パラメータ:

cbmanager (CallbackManager)

戻り値の型:

None