protenix-constraint / model.json
ZhiyuanChen's picture
Upload converted OFoldX checkpoint
5576da3 verified
Raw History Blame Contribute Delete
2.14 kB
{
"architecture": "protenix_constraint",
"atom_feature_size": 385,
"confidence": {
"distance_bin_end": 52.0,
"distance_bin_start": 3.25,
"distance_bin_step": 1.25,
"expansion": 4,
"max_atoms_per_token": 24,
"num_blocks": 4,
"num_pae_bins": 64,
"num_pair_attention_heads": 16,
"num_pde_bins": 64,
"num_plddt_bins": 50,
"num_resolved_bins": 2,
"num_triangle_attention_heads": 4,
"pairformer_dropout": 0.0,
"single_max_abs": 512.0,
"triangle_attention_head_size": 32,
"triangle_multiplication_hidden_size": 128
},
"constraints": true,
"denoiser": {
"atom_hidden_size": 128,
"atom_pair_hidden_size": 16,
"num_atom_decoder_attention_heads": 4,
"num_atom_decoder_blocks": 3,
"num_atom_encoder_attention_heads": 4,
"num_atom_encoder_blocks": 3,
"num_key_atoms": 128,
"num_query_atoms": 32,
"num_token_encoder_attention_heads": 16,
"num_token_encoder_blocks": 24,
"sigma_data": 16.0,
"token_hidden_size": 768
},
"distogram": {
"num_bins": 64
},
"esm": null,
"num_recycles": 10,
"pair_size": 128,
"relative_position_size": 139,
"sampler": {
"churn_rate": 0.8,
"max_noise_level": 160.0,
"min_churn_noise_level": 1.0,
"min_noise_level": 0.0004,
"noise_scale": 1.003,
"num_sampling_steps": 200,
"rho": 7.0,
"solver": "euler",
"step_scale": 1.5
},
"single_size": 384,
"template": null,
"trunk": {
"msa_attention_head_size": null,
"msa_dropout": 0.15,
"msa_expansion": 4,
"msa_outer_product_hidden_size": 32,
"msa_size": 64,
"msa_triangle_attention_head_size": 32,
"msa_triangle_multiplication_hidden_size": 128,
"num_msa_attention_heads": 8,
"num_msa_stages": 4,
"num_msa_triangle_attention_heads": 4,
"num_pair_attention_heads": 16,
"num_pairformer_blocks": 48,
"num_pairformer_triangle_attention_heads": 4,
"pair_dropout": 0.25,
"pairformer_dropout": 0.25,
"pairformer_expansion": 4,
"pairformer_triangle_attention_head_size": 32,
"pairformer_triangle_multiplication_hidden_size": 128
}
}