examples/common_examples/experiment_config.json

"{\n  \"experiment_settings\": {\n    \"mode\": \"predict\",\n    \"output_dir\": \"example_outputs\",\n    \"log_dir\": null,\n    \"seeds\": [\n      42\n    ],\n    \"num_seeds\": null,\n    \"use_msa_server\": true,\n    \"use_templates\": true,\n    \"skip_existing\": false\n  },\n  \"pl_trainer_args\": {\n    \"max_epochs\": 1000,\n    \"accelerator\": \"gpu\",\n    \"precision\": \"32-true\",\n    \"num_nodes\": 1,\n    \"devices\": 4,\n    \"profiler\": null,\n    \"log_every_n_steps\": 1,\n    \"enable_checkpointing\": true,\n    \"enable_model_summary\": false,\n    \"deepspeed_config_path\": null,\n    \"distributed_timeout\": \"PT30M\",\n    \"mpi_plugin\": false\n  },\n  \"model_update\": {\n    \"presets\": [\n      \"predict\",\n      \"pae_enabled\",\n      \"low_mem\"\n    ],\n    \"custom\": {\n      \"settings\": {\n        \"memory\": {\n          \"eval\": {\n            \"use_cueq_triangle_kernels\": false,\n            \"use_deepspeed_evo_attention\": false\n          }\n        }\n      }\n    }\n  },\n  \"inference_ckpt_path\": \"/home/jnwei/.openfold3/of3_ft3_v1.pt\",\n  \"cache_path\": \"/home/jnwei/.openfold3\",\n  \"data_module_args\": {\n    \"batch_size\": 1,\n    \"data_seed\": 42,\n    \"num_workers\": 10,\n    \"num_workers_validation\": 4,\n    \"epoch_len\": 4\n  },\n  \"dataset_config_kwargs\": {\n    \"ccd_file_path\": null,\n    \"msa\": {\n      \"max_rows_paired\": 8191,\n      \"max_rows\": 16384,\n      \"subsample_with_bands\": false,\n      \"min_chains_paired_partial\": 2,\n      \"pairing_mask_keys\": [\n        \"shared_by_two\",\n        \"less_than_600\"\n      ],\n      \"moltypes\": [\n        0,\n        1\n      ],\n      \"max_seq_counts\": {\n        \"uniref90_hits\": 10000,\n        \"uniprot_hits\": 50000,\n        \"bfd_uniclust_hits\": 10000000,\n        \"bfd_uniref_hits\": 10000000,\n        \"cfdb_uniref30\": 10000000,\n        \"mgnify_hits\": 5000,\n        \"rfam_hits\": 10000,\n        \"rnacentral_hits\": 10000,\n        \"nt_hits\": 10000,\n        \"concat_cfdb_uniref100_filtered\": 10000000,\n        \"colabfold_main\": 16384,\n        \"colabfold_paired\": 8192\n      },\n      \"msas_to_pair\": [\n        \"uniprot_hits\",\n        \"uniprot\"\n      ],\n      \"aln_order\": [\n        \"uniref90_hits\",\n        \"bfd_uniclust_hits\",\n        \"bfd_uniref_hits\",\n        \"cfdb_uniref30\",\n        \"mgnify_hits\",\n        \"rfam_hits\",\n        \"rnacentral_hits\",\n        \"nt_hits\",\n        \"concat_cfdb_uniref100_filtered\",\n        \"colabfold_main\"\n      ],\n      \"paired_msa_order\": [\n        \"colabfold_paired\"\n      ]\n    },\n    \"template\": {\n      \"n_templates\": 4,\n      \"take_top_k\": true,\n      \"distogram\": {\n        \"min_bin\": 3.25,\n        \"max_bin\": 50.75,\n        \"n_bins\": 39\n      }\n    }\n  },\n  \"output_writer_settings\": {\n    \"structure_format\": \"cif\",\n    \"full_confidence_output_format\": \"json\",\n    \"write_features\": false,\n    \"write_latent_outputs\": false\n  },\n  \"msa_computation_settings\": {\n    \"msa_file_format\": \"npz\",\n    \"server_user_agent\": \"openfold\",\n    \"server_url\": \"https://api.colabfold.com/\",\n    \"save_mappings\": true,\n    \"msa_output_directory\": \"/var/tmp/pbs.6552323.polaris-pbs-01.hsn.cm.polaris.alcf.anl.gov/of3_colabfold_msas\",\n    \"cleanup_msa_dir\": true\n  },\n  \"template_preprocessor_settings\": {\n    \"mode\": \"predict\",\n    \"moltypes\": [\n      0\n    ],\n    \"max_sequences_parse\": 200,\n    \"max_seq_id\": null,\n    \"min_align\": null,\n    \"min_len\": null,\n    \"max_release_date\": null,\n    \"min_release_date_diff\": null,\n    \"max_templates\": 20,\n    \"fetch_missing_structures\": true,\n    \"create_precache\": false,\n    \"preparse_structures\": false,\n    \"create_logs\": false,\n    \"n_processes\": 1,\n    \"chunksize\": 1,\n    \"structure_directory\": \"/var/tmp/pbs.6552323.polaris-pbs-01.hsn.cm.polaris.alcf.anl.gov/of3_template_data/template_structures\",\n    \"structure_file_format\": \"cif\",\n    \"output_directory\": \"/var/tmp/pbs.6552323.polaris-pbs-01.hsn.cm.polaris.alcf.anl.gov/of3_template_data\",\n    \"precache_directory\": null,\n    \"structure_array_directory\": null,\n    \"cache_directory\": \"/var/tmp/pbs.6552323.polaris-pbs-01.hsn.cm.polaris.alcf.anl.gov/of3_template_data/template_cache\",\n    \"log_directory\": null,\n    \"ccd_file_path\": null\n  }\n}"