Skip to content

GitLab

"...git@developer.sourcefind.cn:OpenDAS/openpcdet.git" did not exist on "e9b06b361c5a65b1ed73aa097ccbacc710b74ae5"
Commit 17015611 authored by Gustaf Ahdritz's avatar Gustaf Ahdritz
Browse files

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

parents 28a4c4d9 32c137e2
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 8 deletions
README.md
View file @ 17015611
...@@ -198,6 +198,7 @@ python3 train_openfold.py mmcif_dir/ alignment_dir/ template_mmcif_dir/ \ ...@@ -198,6 +198,7 @@ python3 train_openfold.py mmcif_dir/ alignment_dir/ template_mmcif_dir/ \
--gpus 8 --replace_sampler_ddp=True \ --gpus 8 --replace_sampler_ddp=True \
--seed 42 \ # in multi-gpu settings, the seed must be specified --seed 42 \ # in multi-gpu settings, the seed must be specified
--deepspeed_config_path deepspeed_config.json \ --deepspeed_config_path deepspeed_config.json \
--checkpoint_every_epoch \
--resume_from_ckpt ckpt_dir/ --resume_from_ckpt ckpt_dir/
``` ```
......
deepspeed_config.json
View file @ 17015611
...@@ -6,14 +6,6 @@ ...@@ -6,14 +6,6 @@
"eps": 1e-05 "eps": 1e-05
} }
}, },
"scheduler": {
"type": "WarmupLR",
"params": {
"warmup_min_lr": 0,
"warmup_max_lr": 0.001,
"warmup_num_steps": 128000
}
},
"fp16": { "fp16": {
"enabled": true, "enabled": true,
"min_loss_scale": 1 "min_loss_scale": 1
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment