Skip to content

GitLab

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

Add checkpointing to README sample command

parent fbec92cb
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
README.md
View file @ 01367e36
...@@ -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/
``` ```
......
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