Update example train/val/test commands

lindawangg · Jun 2, 2022 · 177ae78 · 177ae78
1 parent 5d4f01c
commit 177ae78
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/docs/train_eval_inference.md b/docs/train_eval_inference.md
@@ -32,8 +32,8 @@ python train_tf.py \
     --metaname model.meta \
     --ckptname model \
     --n_classes 2 \
-    --trainfile labels/train_COVIDx9B.txt \
-    --testfile labels/test_COVIDx9B.txt \
+    --trainfile labels/train_COVIDx_CXR-3B.txt \
+    --testfile labels/test_COVIDx_CXR-3B.txt \
     --out_tensorname norm_dense_2/Softmax:0 \
     --logit_tensorname norm_dense_2/MatMul:0
 ```
@@ -51,7 +51,7 @@ python eval.py \
     --metaname model.meta \
     --ckptname model \
     --n_classes 2 \
-    --testfile labels/test_COVIDx9B.txt \
+    --testfile labels/test_COVIDx_CXR-3B.txt \
     --out_tensorname softmax/Softmax:0 \
     --is_medusa_backbone
 ```
@@ -63,7 +63,7 @@ python eval.py \
     --metaname model.meta \
     --ckptname model \
     --n_classes 2 \
-    --testfile labels/test_COVIDx9B.txt \
+    --testfile labels/test_COVIDx_CXR-3B.txt \
     --in_tensorname input_1:0 \
     --out_tensorname norm_dense_2/Softmax:0
 ```