deepfold.utils.checkpoint_utils.map_init_state_dicts

deepfold.utils.checkpoint_utils.map_init_state_dicts(alphafold_state_dict_keys: List[str], init_alphafold_state_dict: dict, init_optimizer_state_dict: dict) → Tuple[dict, dict]