cerebras.modelzoo.common.pytorch_utils.get_checkpoints

cerebras.modelzoo.common.pytorch_utils.get_checkpoints(model_dir)

Gather checkpoints in a model directory.

previous

cerebras.modelzoo.common.pytorch_utils.check_checkpoint_compatibility

next

cerebras.modelzoo.common.pytorch_utils.load_from_checkpoint_file