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

17015611 · Gustaf Ahdritz · 28a4c4d9 · 32c137e2 · 17015611 · 17015611
Commit 17015611 authored Feb 08, 2022 by Gustaf Ahdritz
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 8 deletions

README.md README.md +1 -0

deepspeed_config.json deepspeed_config.json +0 -8

No files found.
--- a/README.md
+++ b/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/
 ```


--- a/deepspeed_config.json
+++ b/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