stanford script fix

chrischoy · Jan 15, 2020 · 75c33f5 · 75c33f5
1 parent fcd9fa4
commit 75c33f5
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 5 deletions.
diff --git 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
@@ -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 \