Update README.md

VMarsocci · Sep 25, 2024 · 2ed5650 · 2ed5650
1 parent 1c0a25f
commit 2ed5650
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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