cerebras.modelzoo.common.pytorch_utils.get_checkpoints

cerebras.modelzoo.common.pytorch_utils.get_checkpoints(model_dir: str) → List[str]

Gather checkpoints in a model directory

previous

cerebras.modelzoo.common.pytorch_utils.check_checkpoint_compatibility

next

cerebras.modelzoo.common.pytorch_utils.is_mup_run