resources/run_config.json

{
    "stage": "initial_training",
    "override_conf": {
      "model": {
        "evoformer_stack": {
          "no_blocks": 8
        }
      },
      "data": {"common": {"max_recycling_iters": 3}, "data_module": {"data_loaders": {"batch_size": 1}}, "train": {"crop_size": 355, "fixed_size": false}},
      "loss": {
        "positions_intra_distogram": {"weight": 0.0},
        "inter_contact": {"weight": 0.05, "pos_class_weight": 50.0},
        "binding_site": {"weight": 0.05},
        "affinity2d": {"weight": 0.03},
        "affinity_cls": {"weight": 0.03},
        "fape_interface": {"weight": 1.0}
      },
      "globals": {"max_lr": 0.0004}
    }
}