From 2ed5650948e7cf953ec224d1704b2a68bc477071 Mon Sep 17 00:00:00 2001 From: Valerio Marsocci <49810041+VMarsocci@users.noreply.github.com> Date: Wed, 25 Sep 2024 14:51:01 +0200 Subject: [PATCH] Update README.md --- README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/README.md b/README.md index 16c30c1f..d37312e7 100644 --- a/README.md +++ b/README.md @@ -529,7 +529,7 @@ An evaluation step is always run after the training. If you want to just run an evaluation, indicate the `eval_dir` where the checkpoints and configurations are stored. ``` -torchrun --nnodes=1 --nproc_per_node=1 run.py --batch_size 1 --eval_dir work-dir/the-folder-where-your-exp-is-saved +torchrun geofm_bench/run.py --config-name=test ckpt_dir=path_to_ckpt_dir ``` ## ✏️ Contributing