cerebras.modelzoo.common.pytorch_utils.check_checkpoint_compatibility

cerebras.modelzoo.common.pytorch_utils.check_checkpoint_compatibility(state_dict: Dict[str, Any])

Checks that the checkpoint is compatible with the current version of modelzoo.

previous

cerebras.modelzoo.common.pytorch_utils

next

cerebras.modelzoo.common.pytorch_utils.get_checkpoints