Merge branch 'main' of github.com:aqlaboratory/openfold into main

README.md

@@ -198,6 +198,7 @@ python3 train_openfold.py mmcif_dir/ alignment_dir/ template_mmcif_dir/ \
     --gpus 8 --replace_sampler_ddp=True \
     --seed 42 \ # in multi-gpu settings, the seed must be specified
     --deepspeed_config_path deepspeed_config.json \
+    --checkpoint_every_epoch \
     --resume_from_ckpt ckpt_dir/
 ```
 

deepspeed_config.json

@@ -6,14 +6,6 @@
       "eps": 1e-05
     }
   },
-  "scheduler": {
-      "type": "WarmupLR",
-      "params": {
-        "warmup_min_lr": 0,
-        "warmup_max_lr": 0.001,
-        "warmup_num_steps": 128000
-      }
-  },
   "fp16": {
     "enabled": true,
     "min_loss_scale": 1