cerebras.modelzoo.common.pytorch_utils.load_from_checkpoint_file

cerebras.modelzoo.common.pytorch_utils.load_from_checkpoint_file(checkpoint_path, check_compatibility=True)

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

previous

cerebras.modelzoo.common.pytorch_utils.get_checkpoints

next

cerebras.modelzoo.common.pytorch_utils.setup_artifact_dir