cerebras.modelzoo.common.pytorch_utils.load_from_checkpoint_file

cerebras.modelzoo.common.pytorch_utils.load_from_checkpoint_file(checkpoint_path: str, check_compatibility: bool = True) → dict

Loads state dict from checkpoint path and checks for version compatibilty.

previous

cerebras.modelzoo.common.pytorch_utils.is_mup_run

next

cerebras.modelzoo.common.pytorch_utils.setup_artifact_dir