Update README.md

zhengbw0324 · Mar 21, 2024 · 97e2cf1 · 97e2cf1
1 parent 826e965
commit 97e2cf1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/README.md b/README.md
@@ -84,7 +84,7 @@ Test with a single GPU:
 ```shell
 DATASET=Games
 DATA_PATH=./data
-OUTPUT_DIR=./ckpt/$DATASET/
+CKPT_PATH=./ckpt/$DATASET/
 RESULTS_FILE=./results/$DATASET/result.json
 
 python test.py \
@@ -104,7 +104,7 @@ Test with multiple GPUs:
 ```shell
 DATASET=Games
 DATA_PATH=./data
-OUTPUT_DIR=./ckpt/$DATASET/
+CKPT_PATH=./ckpt/$DATASET/
 RESULTS_FILE=./results/$DATASET/result.json
 
 torchrun --nproc_per_node=8 --master_port=23324 test_ddp.py \