"tools/git@developer.sourcefind.cn:OpenDAS/openpcdet.git" did not exist on "ba28c08f0c94d39628895d222913d2e09e4c266e"
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:
python3 train_openfold.py mmcif_dir/ alignment_dir/ template_mmcif_dir/ \
2021-10-10 \
--template_release_dates_cache_path mmcif_cache.json \
--precision 16 \
--precision bf16 \
--gpus 8 --replace_sampler_ddp=True \
--seed 42 \ # in multi-gpu settings, the seed must be specified
--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