| { |
| "architecture": "rfdiffusion", |
| "denoiser": { |
| "dropout": 0.15, |
| "freeze_track_motif": false, |
| "full_msa_size": 64, |
| "hidden_size": 32, |
| "msa_size": 256, |
| "num_extra_blocks": 4, |
| "num_main_blocks": 32, |
| "num_msa_attention_heads": 8, |
| "num_pair_attention_heads": 4, |
| "num_refinement_blocks": 4, |
| "num_template_attention_heads": 4, |
| "pair_size": 128, |
| "refinement_se3": { |
| "channel_reduction_factor": 4, |
| "edge_size": 64, |
| "hidden_size": 32, |
| "input_scalar_size": 64, |
| "input_vector_size": 3, |
| "num_attention_heads": 4, |
| "num_blocks": 1, |
| "num_degrees": 2, |
| "output_scalar_size": 64, |
| "output_vector_size": 2 |
| }, |
| "template_hidden_size": 32, |
| "template_size": 64, |
| "trunk_se3": { |
| "channel_reduction_factor": 4, |
| "edge_size": 32, |
| "hidden_size": 32, |
| "input_scalar_size": 8, |
| "input_vector_size": 3, |
| "num_attention_heads": 4, |
| "num_blocks": 1, |
| "num_degrees": 2, |
| "output_scalar_size": 8, |
| "output_vector_size": 2 |
| } |
| }, |
| "diffuser": { |
| "beta_final": 0.07, |
| "beta_initial": 0.01, |
| "coordinate_scale": 0.25, |
| "max_rotation_beta": 2.5, |
| "max_rotation_sigma": 1.5, |
| "min_rotation_beta": 1.5, |
| "min_rotation_sigma": 0.02, |
| "num_timesteps": 50, |
| "rotation_schedule_type": "linear" |
| }, |
| "preprocess": { |
| "motif_sidechain_input": true, |
| "sidechain_input": false, |
| "structure_self_conditioning": true, |
| "template_pair_feature_size": 47, |
| "template_single_feature_size": 28 |
| } |
| } |
|
|
