From 75c33f56b3d00a80a43df03ab0fa4f5ef8f226ad Mon Sep 17 00:00:00 2001
From: Chris Choy <chrischoy@ai.stanford.edu>
Date: Wed, 15 Jan 2020 03:52:53 -0800
Subject: [PATCH] stanford script fix

---
 README.md                 | 4 +---
 scripts/train_stanford.sh | 3 +--
 2 files changed, 2 insertions(+), 5 deletions(-)

diff --git a/README.md b/README.md
index 903b991..0ac08df 100644
--- a/README.md
+++ b/README.md
@@ -81,12 +81,10 @@ python -m lib.datasets.preprocessing.stanford
 
 3. Train
 
-Modify the stanford 3d path in the script and run
-
 ```
 ./scripts/train_stanford.sh 0 \
 	"-default" \
-	""
+	"--stanford3d_path /PATH/TO/PREPROCESSED/STANFORD"
 ```
 
 ## Model Zoo
diff --git a/scripts/train_stanford.sh b/scripts/train_stanford.sh
index 3e8cde6..c614524 100755
--- a/scripts/train_stanford.sh
+++ b/scripts/train_stanford.sh
@@ -21,12 +21,11 @@ mkdir -p $LOG_DIR
 LOG="$LOG_DIR/$TIME.txt"
 
 python main.py \
-    --stanford3d_path ~/datasets/Stanford3D/ \
     --dataset StanfordArea5Dataset \
     --batch_size $BATCH_SIZE \
     --scheduler PolyLR \
     --model Res16UNet18 \
-    --conv1_kernel_size 5
+    --conv1_kernel_size 5 \
     --log_dir $LOG_DIR \
     --lr 1e-1 \
     --max_iter 60000 \