Unverified Commit ce2e1f29 authored by Gustaf Ahdritz's avatar Gustaf Ahdritz Committed by GitHub
Browse files

Fix sample training command

parent f3087e31
...@@ -295,7 +295,7 @@ Finally, call the training script: ...@@ -295,7 +295,7 @@ Finally, call the training script:
python3 train_openfold.py mmcif_dir/ alignment_dir/ template_mmcif_dir/ \ python3 train_openfold.py mmcif_dir/ alignment_dir/ template_mmcif_dir/ \
2021-10-10 \ 2021-10-10 \
--template_release_dates_cache_path mmcif_cache.json \ --template_release_dates_cache_path mmcif_cache.json \
--precision 16 \ --precision bf16 \
--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 \
......
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