| { | |
| "CONFIG": { | |
| "data": { | |
| "transform": { | |
| "common": { | |
| "crop_size": { | |
| "eval": null, | |
| "finetune": 384, | |
| "predict": null, | |
| "train": 256 | |
| }, | |
| "feats": { | |
| "aatype": [ | |
| "num residues placeholder" | |
| ], | |
| "all_atom_mask": [ | |
| "num residues placeholder", | |
| null | |
| ], | |
| "all_atom_positions": [ | |
| "num residues placeholder", | |
| null, | |
| null | |
| ], | |
| "alt_chi_angles": [ | |
| "num residues placeholder", | |
| null | |
| ], | |
| "assembly_num_chains": [ | |
| null | |
| ], | |
| "asym_id": [ | |
| "num residues placeholder" | |
| ], | |
| "asym_len": [ | |
| null | |
| ], | |
| "atom14_alt_gt_exists": [ | |
| "num residues placeholder", | |
| null | |
| ], | |
| "atom14_alt_gt_positions": [ | |
| "num residues placeholder", | |
| null, | |
| null | |
| ], | |
| "atom14_atom_exists": [ | |
| "num residues placeholder", | |
| null | |
| ], | |
| "atom14_atom_is_ambiguous": [ | |
| "num residues placeholder", | |
| null | |
| ], | |
| "atom14_gt_exists": [ | |
| "num residues placeholder", | |
| null | |
| ], | |
| "atom14_gt_positions": [ | |
| "num residues placeholder", | |
| null, | |
| null | |
| ], | |
| "atom37_atom_exists": [ | |
| "num residues placeholder", | |
| null | |
| ], | |
| "backbone_rigid_mask": [ | |
| "num residues placeholder" | |
| ], | |
| "backbone_rigid_tensor": [ | |
| "num residues placeholder", | |
| null, | |
| null | |
| ], | |
| "bert_mask": [ | |
| "msa placeholder", | |
| "num residues placeholder" | |
| ], | |
| "chi_angles_sin_cos": [ | |
| "num residues placeholder", | |
| null, | |
| null | |
| ], | |
| "chi_mask": [ | |
| "num residues placeholder", | |
| null | |
| ], | |
| "cluster_bias_mask": [ | |
| "msa placeholder" | |
| ], | |
| "entity_id": [ | |
| "num residues placeholder" | |
| ], | |
| "extra_deletion_value": [ | |
| "extra msa placeholder", | |
| "num residues placeholder" | |
| ], | |
| "extra_has_deletion": [ | |
| "extra msa placeholder", | |
| "num residues placeholder" | |
| ], | |
| "extra_msa": [ | |
| "extra msa placeholder", | |
| "num residues placeholder" | |
| ], | |
| "extra_msa_mask": [ | |
| "extra msa placeholder", | |
| "num residues placeholder" | |
| ], | |
| "extra_msa_row_mask": [ | |
| "extra msa placeholder" | |
| ], | |
| "is_distillation": [], | |
| "msa_feat": [ | |
| "msa placeholder", | |
| "num residues placeholder", | |
| null | |
| ], | |
| "msa_mask": [ | |
| "msa placeholder", | |
| "num residues placeholder" | |
| ], | |
| "msa_row_mask": [ | |
| "msa placeholder" | |
| ], | |
| "no_recycling_iters": [], | |
| "num_sym": [ | |
| "num residues placeholder" | |
| ], | |
| "pseudo_beta": [ | |
| "num residues placeholder", | |
| null | |
| ], | |
| "pseudo_beta_mask": [ | |
| "num residues placeholder" | |
| ], | |
| "residue_index": [ | |
| "num residues placeholder" | |
| ], | |
| "residx_atom14_to_atom37": [ | |
| "num residues placeholder", | |
| null | |
| ], | |
| "residx_atom37_to_atom14": [ | |
| "num residues placeholder", | |
| null | |
| ], | |
| "resolution": [], | |
| "rigidgroups_alt_gt_frames": [ | |
| "num residues placeholder", | |
| null, | |
| null, | |
| null | |
| ], | |
| "rigidgroups_group_exists": [ | |
| "num residues placeholder", | |
| null | |
| ], | |
| "rigidgroups_group_is_ambiguous": [ | |
| "num residues placeholder", | |
| null | |
| ], | |
| "rigidgroups_gt_exists": [ | |
| "num residues placeholder", | |
| null | |
| ], | |
| "rigidgroups_gt_frames": [ | |
| "num residues placeholder", | |
| null, | |
| null, | |
| null | |
| ], | |
| "seq_length": [], | |
| "seq_mask": [ | |
| "num residues placeholder" | |
| ], | |
| "sym_id": [ | |
| "num residues placeholder" | |
| ], | |
| "target_feat": [ | |
| "num residues placeholder", | |
| null | |
| ], | |
| "template_aatype": [ | |
| "num templates placeholder", | |
| "num residues placeholder" | |
| ], | |
| "template_all_atom_mask": [ | |
| "num templates placeholder", | |
| "num residues placeholder", | |
| null | |
| ], | |
| "template_all_atom_positions": [ | |
| "num templates placeholder", | |
| "num residues placeholder", | |
| null, | |
| null | |
| ], | |
| "template_alt_torsion_angles_sin_cos": [ | |
| "num templates placeholder", | |
| "num residues placeholder", | |
| null, | |
| null | |
| ], | |
| "template_backbone_rigid_mask": [ | |
| "num templates placeholder", | |
| "num residues placeholder" | |
| ], | |
| "template_backbone_rigid_tensor": [ | |
| "num templates placeholder", | |
| "num residues placeholder", | |
| null, | |
| null | |
| ], | |
| "template_mask": [ | |
| "num templates placeholder" | |
| ], | |
| "template_pseudo_beta": [ | |
| "num templates placeholder", | |
| "num residues placeholder", | |
| null | |
| ], | |
| "template_pseudo_beta_mask": [ | |
| "num templates placeholder", | |
| "num residues placeholder" | |
| ], | |
| "template_sum_probs": [ | |
| "num templates placeholder", | |
| null | |
| ], | |
| "template_torsion_angles_mask": [ | |
| "num templates placeholder", | |
| "num residues placeholder", | |
| null | |
| ], | |
| "template_torsion_angles_sin_cos": [ | |
| "num templates placeholder", | |
| "num residues placeholder", | |
| null, | |
| null | |
| ], | |
| "true_msa": [ | |
| "msa placeholder", | |
| "num residues placeholder" | |
| ], | |
| "use_clamped_fape": [] | |
| }, | |
| "max_extra_msa": { | |
| "eval": 512, | |
| "finetune": 5120, | |
| "predict": 1024, | |
| "train": 1024 | |
| }, | |
| "max_msa_clusters": { | |
| "eval": 128, | |
| "finetune": 512, | |
| "predict": 512, | |
| "train": 128 | |
| }, | |
| "max_templates": 4, | |
| "no_recycling_iters": 3, | |
| "reduce_msa_clusters_by_max_templates": false, | |
| "resample_msa_in_recycling": true, | |
| "subsample_templates": { | |
| "eval": false, | |
| "finetune": true, | |
| "predict": false, | |
| "train": true | |
| }, | |
| "supervised": { | |
| "eval": true, | |
| "finetune": true, | |
| "predict": false, | |
| "train": true | |
| }, | |
| "supervised_features": [ | |
| "all_atom_mask", | |
| "all_atom_positions", | |
| "resolution", | |
| "use_clamped_fape", | |
| "is_distillation" | |
| ], | |
| "template_features": [ | |
| "template_all_atom_positions", | |
| "template_sum_probs", | |
| "template_aatype", | |
| "template_all_atom_mask" | |
| ], | |
| "uniform_recycling": { | |
| "eval": false, | |
| "finetune": true, | |
| "predict": false, | |
| "train": true | |
| }, | |
| "unsupervised_features": [ | |
| "aatype", | |
| "residue_index", | |
| "msa", | |
| "num_alignments", | |
| "seq_length", | |
| "between_segment_residues", | |
| "deletion_matrix", | |
| "no_recycling_iters" | |
| ] | |
| }, | |
| "make_masked_msa": { | |
| "profile_prob": 0.1, | |
| "replace_fraction": 0.15, | |
| "same_prob": 0.1, | |
| "uniform_prob": 0.1 | |
| }, | |
| "make_relative_encoding_feat": { | |
| "max_relative_feature": 32 | |
| }, | |
| "nearest_neighbor_clusters": { | |
| "gap_agreement_weight": 0.0 | |
| }, | |
| "randomly_replace_msa_with_unknown": { | |
| "replace_proportion": 0.0 | |
| }, | |
| "sample_msa": { | |
| "reduce_msa_clusters_by_max_templates": false | |
| } | |
| } | |
| }, | |
| "globals": { | |
| "is_multimer": false, | |
| "mode": "predict", | |
| "num_recycles": 3, | |
| "recycle_early_stop_tolerance": 0.5, | |
| "use_templates": false | |
| }, | |
| "model": { | |
| "auxiliary_heads": { | |
| "c_m": 256, | |
| "c_s": 384, | |
| "c_z": 128, | |
| "distogram_head_num_bins": 64, | |
| "experimentally_resolved_head_c_out": 37, | |
| "masked_msa_head_c_out": 23, | |
| "predicted_aligned_error_head_num_bins": 64, | |
| "predicted_lddt_head_c_hidden": 128, | |
| "predicted_lddt_head_num_bins": 50, | |
| "tm_enabled": false | |
| }, | |
| "evoformer_stack": { | |
| "c_h_m": 32, | |
| "c_h_opm": 32, | |
| "c_h_u": 128, | |
| "c_h_z": 32, | |
| "c_m": 256, | |
| "c_s": 384, | |
| "c_z": 128, | |
| "fused": false, | |
| "h_m": 8, | |
| "h_z": 4, | |
| "msa_global_col_attention": false, | |
| "n_trans": 4, | |
| "num_blocks": 48, | |
| "opm_first": false, | |
| "ouput_single": true | |
| }, | |
| "extra_msa_stack": { | |
| "c_h_m": 8, | |
| "c_h_opm": 32, | |
| "c_h_u": 128, | |
| "c_h_z": 32, | |
| "c_m": 64, | |
| "c_z": 128, | |
| "fused": false, | |
| "h_m": 8, | |
| "h_z": 4, | |
| "msa_global_col_attention": true, | |
| "n_trans": 4, | |
| "num_blocks": 4, | |
| "opm_first": false | |
| }, | |
| "feature_embedder": { | |
| "c_e": 64, | |
| "c_m": 256, | |
| "c_t": 64, | |
| "c_taf": 57, | |
| "c_tpf": 88, | |
| "c_z": 128, | |
| "extra_msa_dim": 25, | |
| "max_relative_feature": 32, | |
| "msa_dim": 49, | |
| "num_bins": 15, | |
| "tf_dim": 22 | |
| }, | |
| "loss": { | |
| "distogram_loss": { | |
| "eps": 1e-06, | |
| "max_bin": 21.6875, | |
| "min_bin": 2.3125, | |
| "no_bins": 64, | |
| "weight": 0.3 | |
| }, | |
| "experimentally_resolved_loss": { | |
| "eps": 1e-08, | |
| "max_resolution": 3.0, | |
| "min_resolution": 0.1, | |
| "weight": 0.0 | |
| }, | |
| "fape_loss": { | |
| "backbone": { | |
| "clamp_distance": 10.0, | |
| "loss_unit_distance": 10.0, | |
| "weight": 0.5 | |
| }, | |
| "eps": 0.0001, | |
| "sidechain": { | |
| "clamp_distance": 10.0, | |
| "length_scale": 10.0, | |
| "weight": 0.5 | |
| }, | |
| "weight": 1.0 | |
| }, | |
| "masked_msa_loss": { | |
| "eps": 1e-08, | |
| "weight": 2.0 | |
| }, | |
| "predicted_lddt_loss": { | |
| "cutoff": 15.0, | |
| "eps": 1e-10, | |
| "max_resolution": 3.0, | |
| "min_resolution": 0.1, | |
| "no_bins": 50, | |
| "weight": 0.01 | |
| }, | |
| "predicted_tm_loss": { | |
| "eps": 1e-08, | |
| "max_bin": 31, | |
| "max_resolution": 3.0, | |
| "min_resolution": 0.1, | |
| "no_bins": 64, | |
| "weight": 0.1 | |
| }, | |
| "supervised_chi_loss": { | |
| "angle_norm_weight": 0.01, | |
| "chi_weight": 0.5, | |
| "eps": 1e-06, | |
| "weight": 1.0 | |
| }, | |
| "violation_loss": { | |
| "clash_overlap_tolerance": 1.5, | |
| "eps": 1e-06, | |
| "violation_tolerance_factor": 12.0, | |
| "weight": 0.0 | |
| } | |
| }, | |
| "structure_module": { | |
| "c_s": 384, | |
| "c_sc": 128, | |
| "c_z": 128, | |
| "dropout": 0.1, | |
| "ipa_scalar_bias": true, | |
| "no_angles": 7, | |
| "num_head": 12, | |
| "num_layer": 8, | |
| "num_point_qk": 4, | |
| "num_point_v": 8, | |
| "num_scalar_qk": 16, | |
| "num_scalar_v": 16, | |
| "position_scale": 10.0, | |
| "separate_kv": false | |
| }, | |
| "template_pair_stack": { | |
| "attention_first": true, | |
| "c_h_t": 16, | |
| "c_h_u": 64, | |
| "c_t": 64, | |
| "c_z": 128, | |
| "fused": false, | |
| "h_t": 4, | |
| "n_trans": 2, | |
| "num_blocks": 2 | |
| } | |
| } | |
| }, | |
| "CONFIG_DIFFS": { | |
| "model_1": { | |
| "data.transform.common.max_extra_msa.predict": 5120, | |
| "data.transform.common.max_extra_msa.train": 5120, | |
| "data.transform.use_templates": true, | |
| "globals.use_templates": true | |
| }, | |
| "model_1_multimer": { | |
| "data.transform.common.max_extra_msa": { | |
| "eval": 1152, | |
| "finetune": 1152, | |
| "predict": 1152, | |
| "train": 1152 | |
| }, | |
| "data.transform.common.max_msa_clusters": { | |
| "eval": 128, | |
| "finetune": 252, | |
| "predict": 128, | |
| "train": 128 | |
| }, | |
| "globals.is_multimer": true, | |
| "globals.use_templates": true, | |
| "model.auxiliary_heads.masked_msa_head_c_out": 22, | |
| "model.auxiliary_heads.tm_enabled": true, | |
| "model.evoformer_stack.opm_first": true, | |
| "model.extra_msa_stack.opm_first": true, | |
| "model.feature_embedder.c_taf": 34, | |
| "model.feature_embedder.tf_dim": 21, | |
| "model.loss.supervised_chi_loss.weight": 3.0, | |
| "model.structure_module.ipa_scalar_bias": false, | |
| "model.structure_module.position_scale": 20, | |
| "model.structure_module.separate_kv": true, | |
| "model.template_pair_stack.attention_first": false | |
| }, | |
| "model_1_multimer_v2": { | |
| "data.transform.common.max_extra_msa": { | |
| "eval": 1152, | |
| "finetune": 1152, | |
| "predict": 1152, | |
| "train": 1152 | |
| }, | |
| "data.transform.common.max_msa_clusters": { | |
| "eval": 128, | |
| "finetune": 512, | |
| "predict": 252, | |
| "train": 128 | |
| }, | |
| "globals.is_multimer": true, | |
| "globals.use_templates": true, | |
| "model.auxiliary_heads.masked_msa_head_c_out": 22, | |
| "model.auxiliary_heads.tm_enabled": true, | |
| "model.evoformer_stack.opm_first": true, | |
| "model.extra_msa_stack.opm_first": true, | |
| "model.feature_embedder.c_taf": 34, | |
| "model.feature_embedder.tf_dim": 21, | |
| "model.loss.supervised_chi_loss.weight": 3.0, | |
| "model.structure_module.ipa_scalar_bias": false, | |
| "model.structure_module.position_scale": 20, | |
| "model.structure_module.separate_kv": true, | |
| "model.template_pair_stack.attention_first": false | |
| }, | |
| "model_1_multimer_v3": { | |
| "data.transform.common.max_extra_msa": { | |
| "eval": 1152, | |
| "finetune": 1152, | |
| "predict": 2048, | |
| "train": 1152 | |
| }, | |
| "data.transform.common.max_msa_clusters": { | |
| "eval": 128, | |
| "finetune": 512, | |
| "predict": 508, | |
| "train": 128 | |
| }, | |
| "globals.is_multimer": true, | |
| "globals.use_templates": true, | |
| "model.auxiliary_heads.masked_msa_head_c_out": 22, | |
| "model.auxiliary_heads.tm_enabled": true, | |
| "model.evoformer_stack.fused": true, | |
| "model.evoformer_stack.opm_first": true, | |
| "model.extra_msa_stack.fused": true, | |
| "model.extra_msa_stack.opm_first": true, | |
| "model.feature_embedder.c_taf": 34, | |
| "model.feature_embedder.tf_dim": 21, | |
| "model.structure_module.ipa_scalar_bias": false, | |
| "model.structure_module.position_scale": 20, | |
| "model.structure_module.separate_kv": true, | |
| "model.template_pair_stack.attention_first": false | |
| }, | |
| "model_1_ptm": { | |
| "data.transform.common.max_extra_msa.predict": 5120, | |
| "data.transform.common.max_extra_msa.train": 5120, | |
| "data.transform.use_templates": true, | |
| "globals.use_templates": true, | |
| "model.auxiliary_heads.tm_enabled": true | |
| }, | |
| "model_2": { | |
| "globals.use_templates": true | |
| }, | |
| "model_2_multimer": { | |
| "data.transform.common.max_extra_msa": { | |
| "eval": 1152, | |
| "finetune": 1152, | |
| "predict": 1152, | |
| "train": 1152 | |
| }, | |
| "data.transform.common.max_msa_clusters": { | |
| "eval": 128, | |
| "finetune": 252, | |
| "predict": 128, | |
| "train": 128 | |
| }, | |
| "globals.is_multimer": true, | |
| "globals.use_templates": true, | |
| "model.auxiliary_heads.masked_msa_head_c_out": 22, | |
| "model.auxiliary_heads.tm_enabled": true, | |
| "model.evoformer_stack.opm_first": true, | |
| "model.extra_msa_stack.opm_first": true, | |
| "model.feature_embedder.c_taf": 34, | |
| "model.feature_embedder.tf_dim": 21, | |
| "model.loss.supervised_chi_loss.weight": 3.0, | |
| "model.structure_module.ipa_scalar_bias": false, | |
| "model.structure_module.position_scale": 20, | |
| "model.structure_module.separate_kv": true, | |
| "model.template_pair_stack.attention_first": false | |
| }, | |
| "model_2_multimer_v2": { | |
| "data.transform.common.max_extra_msa": { | |
| "eval": 1152, | |
| "finetune": 1152, | |
| "predict": 1152, | |
| "train": 1152 | |
| }, | |
| "data.transform.common.max_msa_clusters": { | |
| "eval": 128, | |
| "finetune": 512, | |
| "predict": 252, | |
| "train": 128 | |
| }, | |
| "globals.is_multimer": true, | |
| "globals.use_templates": true, | |
| "model.auxiliary_heads.masked_msa_head_c_out": 22, | |
| "model.auxiliary_heads.tm_enabled": true, | |
| "model.evoformer_stack.opm_first": true, | |
| "model.extra_msa_stack.opm_first": true, | |
| "model.feature_embedder.c_taf": 34, | |
| "model.feature_embedder.tf_dim": 21, | |
| "model.loss.supervised_chi_loss.weight": 3.0, | |
| "model.structure_module.ipa_scalar_bias": false, | |
| "model.structure_module.position_scale": 20, | |
| "model.structure_module.separate_kv": true, | |
| "model.template_pair_stack.attention_first": false | |
| }, | |
| "model_2_multimer_v3": { | |
| "data.transform.common.max_extra_msa": { | |
| "eval": 1152, | |
| "finetune": 1152, | |
| "predict": 2048, | |
| "train": 1152 | |
| }, | |
| "data.transform.common.max_msa_clusters": { | |
| "eval": 128, | |
| "finetune": 512, | |
| "predict": 508, | |
| "train": 128 | |
| }, | |
| "globals.is_multimer": true, | |
| "globals.use_templates": true, | |
| "model.auxiliary_heads.masked_msa_head_c_out": 22, | |
| "model.auxiliary_heads.tm_enabled": true, | |
| "model.evoformer_stack.fused": true, | |
| "model.evoformer_stack.opm_first": true, | |
| "model.extra_msa_stack.fused": true, | |
| "model.extra_msa_stack.opm_first": true, | |
| "model.feature_embedder.c_taf": 34, | |
| "model.feature_embedder.tf_dim": 21, | |
| "model.structure_module.ipa_scalar_bias": false, | |
| "model.structure_module.position_scale": 20, | |
| "model.structure_module.separate_kv": true, | |
| "model.template_pair_stack.attention_first": false | |
| }, | |
| "model_2_ptm": { | |
| "globals.use_templates": true, | |
| "model.auxiliary_heads.tm_enabled": true | |
| }, | |
| "model_3": { | |
| "data.transform.common.max_extra_msa.predict": 5120, | |
| "data.transform.common.max_extra_msa.train": 5120 | |
| }, | |
| "model_3_multimer": { | |
| "data.transform.common.max_extra_msa": { | |
| "eval": 1152, | |
| "finetune": 1152, | |
| "predict": 1152, | |
| "train": 1152 | |
| }, | |
| "data.transform.common.max_msa_clusters": { | |
| "eval": 128, | |
| "finetune": 252, | |
| "predict": 128, | |
| "train": 128 | |
| }, | |
| "globals.is_multimer": true, | |
| "globals.use_templates": true, | |
| "model.auxiliary_heads.masked_msa_head_c_out": 22, | |
| "model.auxiliary_heads.tm_enabled": true, | |
| "model.evoformer_stack.opm_first": true, | |
| "model.extra_msa_stack.opm_first": true, | |
| "model.feature_embedder.c_taf": 34, | |
| "model.feature_embedder.tf_dim": 21, | |
| "model.loss.supervised_chi_loss.weight": 3.0, | |
| "model.structure_module.ipa_scalar_bias": false, | |
| "model.structure_module.position_scale": 20, | |
| "model.structure_module.separate_kv": true, | |
| "model.template_pair_stack.attention_first": false | |
| }, | |
| "model_3_multimer_v2": { | |
| "data.transform.common.max_extra_msa": { | |
| "eval": 1152, | |
| "finetune": 1152, | |
| "predict": 1152, | |
| "train": 1152 | |
| }, | |
| "data.transform.common.max_msa_clusters": { | |
| "eval": 128, | |
| "finetune": 512, | |
| "predict": 252, | |
| "train": 128 | |
| }, | |
| "globals.is_multimer": true, | |
| "globals.use_templates": true, | |
| "model.auxiliary_heads.masked_msa_head_c_out": 22, | |
| "model.auxiliary_heads.tm_enabled": true, | |
| "model.evoformer_stack.opm_first": true, | |
| "model.extra_msa_stack.opm_first": true, | |
| "model.feature_embedder.c_taf": 34, | |
| "model.feature_embedder.tf_dim": 21, | |
| "model.loss.supervised_chi_loss.weight": 3.0, | |
| "model.structure_module.ipa_scalar_bias": false, | |
| "model.structure_module.position_scale": 20, | |
| "model.structure_module.separate_kv": true, | |
| "model.template_pair_stack.attention_first": false | |
| }, | |
| "model_3_multimer_v3": { | |
| "data.transform.common.max_extra_msa": { | |
| "eval": 1152, | |
| "finetune": 1152, | |
| "predict": 2048, | |
| "train": 1152 | |
| }, | |
| "data.transform.common.max_msa_clusters": { | |
| "eval": 128, | |
| "finetune": 512, | |
| "predict": 508, | |
| "train": 128 | |
| }, | |
| "globals.is_multimer": true, | |
| "globals.use_templates": true, | |
| "model.auxiliary_heads.masked_msa_head_c_out": 22, | |
| "model.auxiliary_heads.tm_enabled": true, | |
| "model.evoformer_stack.fused": true, | |
| "model.evoformer_stack.opm_first": true, | |
| "model.extra_msa_stack.fused": true, | |
| "model.extra_msa_stack.opm_first": true, | |
| "model.feature_embedder.c_taf": 34, | |
| "model.feature_embedder.tf_dim": 21, | |
| "model.structure_module.ipa_scalar_bias": false, | |
| "model.structure_module.position_scale": 20, | |
| "model.structure_module.separate_kv": true, | |
| "model.template_pair_stack.attention_first": false | |
| }, | |
| "model_3_ptm": { | |
| "data.transform.common.max_extra_msa.predict": 5120, | |
| "data.transform.common.max_extra_msa.train": 5120, | |
| "model.auxiliary_heads.tm_enabled": true | |
| }, | |
| "model_4": { | |
| "data.transform.common.max_extra_msa.predict": 5120, | |
| "data.transform.common.max_extra_msa.train": 5120 | |
| }, | |
| "model_4_multimer": { | |
| "data.transform.common.max_extra_msa": { | |
| "eval": 1152, | |
| "finetune": 1152, | |
| "predict": 1152, | |
| "train": 1152 | |
| }, | |
| "data.transform.common.max_msa_clusters": { | |
| "eval": 128, | |
| "finetune": 252, | |
| "predict": 128, | |
| "train": 128 | |
| }, | |
| "globals.is_multimer": true, | |
| "globals.use_templates": true, | |
| "model.auxiliary_heads.masked_msa_head_c_out": 22, | |
| "model.auxiliary_heads.tm_enabled": true, | |
| "model.evoformer_stack.opm_first": true, | |
| "model.extra_msa_stack.opm_first": true, | |
| "model.feature_embedder.c_taf": 34, | |
| "model.feature_embedder.tf_dim": 21, | |
| "model.loss.supervised_chi_loss.weight": 3.0, | |
| "model.structure_module.ipa_scalar_bias": false, | |
| "model.structure_module.position_scale": 20, | |
| "model.structure_module.separate_kv": true, | |
| "model.template_pair_stack.attention_first": false | |
| }, | |
| "model_4_multimer_v2": { | |
| "data.transform.common.max_extra_msa": { | |
| "eval": 1152, | |
| "finetune": 1152, | |
| "predict": 1152, | |
| "train": 1152 | |
| }, | |
| "data.transform.common.max_msa_clusters": { | |
| "eval": 128, | |
| "finetune": 512, | |
| "predict": 252, | |
| "train": 128 | |
| }, | |
| "globals.is_multimer": true, | |
| "globals.use_templates": true, | |
| "model.auxiliary_heads.masked_msa_head_c_out": 22, | |
| "model.auxiliary_heads.tm_enabled": true, | |
| "model.evoformer_stack.opm_first": true, | |
| "model.extra_msa_stack.opm_first": true, | |
| "model.feature_embedder.c_taf": 34, | |
| "model.feature_embedder.tf_dim": 21, | |
| "model.loss.supervised_chi_loss.weight": 3.0, | |
| "model.structure_module.ipa_scalar_bias": false, | |
| "model.structure_module.position_scale": 20, | |
| "model.structure_module.separate_kv": true, | |
| "model.template_pair_stack.attention_first": false | |
| }, | |
| "model_4_multimer_v3": { | |
| "data.transform.common.max_extra_msa": { | |
| "eval": 1152, | |
| "finetune": 1152, | |
| "predict": 1152, | |
| "train": 1152 | |
| }, | |
| "data.transform.common.max_msa_clusters": { | |
| "eval": 128, | |
| "finetune": 512, | |
| "predict": 508, | |
| "train": 128 | |
| }, | |
| "globals.is_multimer": true, | |
| "globals.use_templates": true, | |
| "model.auxiliary_heads.masked_msa_head_c_out": 22, | |
| "model.auxiliary_heads.tm_enabled": true, | |
| "model.evoformer_stack.fused": true, | |
| "model.evoformer_stack.opm_first": true, | |
| "model.extra_msa_stack.fused": true, | |
| "model.extra_msa_stack.opm_first": true, | |
| "model.feature_embedder.c_taf": 34, | |
| "model.feature_embedder.tf_dim": 21, | |
| "model.structure_module.ipa_scalar_bias": false, | |
| "model.structure_module.position_scale": 20, | |
| "model.structure_module.separate_kv": true, | |
| "model.template_pair_stack.attention_first": false | |
| }, | |
| "model_4_ptm": { | |
| "data.transform.common.max_extra_msa.predict": 5120, | |
| "data.transform.common.max_extra_msa.train": 5120, | |
| "model.auxiliary_heads.tm_enabled": true | |
| }, | |
| "model_5": {}, | |
| "model_5_multimer": { | |
| "data.transform.common.max_extra_msa": { | |
| "eval": 1152, | |
| "finetune": 1152, | |
| "predict": 1152, | |
| "train": 1152 | |
| }, | |
| "data.transform.common.max_msa_clusters": { | |
| "eval": 128, | |
| "finetune": 252, | |
| "predict": 128, | |
| "train": 128 | |
| }, | |
| "globals.is_multimer": true, | |
| "globals.use_templates": true, | |
| "model.auxiliary_heads.masked_msa_head_c_out": 22, | |
| "model.auxiliary_heads.tm_enabled": true, | |
| "model.evoformer_stack.opm_first": true, | |
| "model.extra_msa_stack.opm_first": true, | |
| "model.feature_embedder.c_taf": 34, | |
| "model.feature_embedder.tf_dim": 21, | |
| "model.loss.supervised_chi_loss.weight": 3.0, | |
| "model.structure_module.ipa_scalar_bias": false, | |
| "model.structure_module.position_scale": 20, | |
| "model.structure_module.separate_kv": true, | |
| "model.template_pair_stack.attention_first": false | |
| }, | |
| "model_5_multimer_v2": { | |
| "data.transform.common.max_extra_msa": { | |
| "eval": 1152, | |
| "finetune": 1152, | |
| "predict": 1152, | |
| "train": 1152 | |
| }, | |
| "data.transform.common.max_msa_clusters": { | |
| "eval": 128, | |
| "finetune": 512, | |
| "predict": 252, | |
| "train": 128 | |
| }, | |
| "globals.is_multimer": true, | |
| "globals.use_templates": true, | |
| "model.auxiliary_heads.masked_msa_head_c_out": 22, | |
| "model.auxiliary_heads.tm_enabled": true, | |
| "model.evoformer_stack.opm_first": true, | |
| "model.extra_msa_stack.opm_first": true, | |
| "model.feature_embedder.c_taf": 34, | |
| "model.feature_embedder.tf_dim": 21, | |
| "model.loss.supervised_chi_loss.weight": 3.0, | |
| "model.structure_module.ipa_scalar_bias": false, | |
| "model.structure_module.position_scale": 20, | |
| "model.structure_module.separate_kv": true, | |
| "model.template_pair_stack.attention_first": false | |
| }, | |
| "model_5_multimer_v3": { | |
| "data.transform.common.max_extra_msa": { | |
| "eval": 1152, | |
| "finetune": 1152, | |
| "predict": 1152, | |
| "train": 1152 | |
| }, | |
| "data.transform.common.max_msa_clusters": { | |
| "eval": 128, | |
| "finetune": 512, | |
| "predict": 508, | |
| "train": 128 | |
| }, | |
| "globals.is_multimer": true, | |
| "globals.use_templates": true, | |
| "model.auxiliary_heads.masked_msa_head_c_out": 22, | |
| "model.auxiliary_heads.tm_enabled": true, | |
| "model.evoformer_stack.fused": true, | |
| "model.evoformer_stack.opm_first": true, | |
| "model.extra_msa_stack.fused": true, | |
| "model.extra_msa_stack.opm_first": true, | |
| "model.feature_embedder.c_taf": 34, | |
| "model.feature_embedder.tf_dim": 21, | |
| "model.structure_module.ipa_scalar_bias": false, | |
| "model.structure_module.position_scale": 20, | |
| "model.structure_module.separate_kv": true, | |
| "model.template_pair_stack.attention_first": false | |
| }, | |
| "model_5_ptm": { | |
| "model.auxiliary_heads.tm_enabled": true | |
| } | |
| }, | |
| "_config": null, | |
| "architectures": [ | |
| "STFold" | |
| ], | |
| "mode": "predict", | |
| "model_type": "STFold", | |
| "stfold_type": "model_3_multimer", | |
| "torch_dtype": "float32", | |
| "transformers_version": "4.31.0.dev0" | |
| } | |