cerebras.modelzoo.common.pytorch_utils.load_from_checkpoint_file#

cerebras.modelzoo.common.pytorch_utils.load_from_checkpoint_file(checkpoint_path, check_compatibility=True)[source]#

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