File size: 1,237 Bytes
b92557c | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 | {
"architecture": "pxdesign",
"atom_feature_size": 385,
"condition_embeddings": {
"add_feature_size": 4,
"atom_hidden_size": 128,
"atom_pair_hidden_size": 16,
"design_feature_size": 1,
"hotspot_size": 1,
"num_atom_encoder_attention_heads": 4,
"num_atom_encoder_blocks": 3,
"num_key_atoms": 128,
"num_query_atoms": 32,
"plddt_size": 1,
"token_hidden_size": 384,
"vocab_size": 72
},
"denoiser": {
"atom_hidden_size": 128,
"atom_pair_hidden_size": 16,
"num_atom_decoder_attention_heads": 4,
"num_atom_decoder_blocks": 4,
"num_atom_encoder_attention_heads": 4,
"num_atom_encoder_blocks": 4,
"num_key_atoms": 128,
"num_query_atoms": 32,
"num_token_encoder_attention_heads": 16,
"num_token_encoder_blocks": 16,
"sigma_data": 16.0,
"token_hidden_size": 768
},
"num_template_bins": 64,
"pair_size": 128,
"sampler": {
"churn_rate": 1.0,
"max_noise_level": 160.0,
"max_step_scale": 2.5,
"min_churn_noise_level": 0.01,
"min_noise_level": 0.0004,
"min_step_scale": 1.0,
"noise_scale": 1.003,
"rho": 7.0,
"step_scale_schedule": "piecewise_65"
},
"single_size": 384,
"token_feature_size": 449
}
|