braintools.trainer.list_checkpoints

braintools.trainer.list_checkpoints(dirpath, pattern='*.ckpt', sort_by='time')

List all checkpoints in a directory.

Parameters:

• dirpath (str) – Directory to search in.

• pattern (str) – Glob pattern for checkpoint files.

• sort_by (str) – How to sort checkpoints (‘time’, ‘name’, ‘epoch’).

Returns:

List of checkpoint file paths.

Return type:

List[str]

Examples

>>> checkpoints = list_checkpoints('checkpoints/')
>>> for ckpt in checkpoints:
...     print(ckpt)