Modify config

tum-phoenix · Jul 17, 2019 · fce4045 · fce4045
1 parent 17251aa
commit fce4045
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/ENet-Label-Torch/experiments/train.sh b/ENet-Label-Torch/experiments/train.sh
@@ -1,16 +1,16 @@
 #!/usr/bin/env sh
 exp=vgg_SCNN_DULR_w9
 data=./data/CULane
-CUDA_VISIBLE_DEVICES="0,1" th main.lua \
+CUDA_VISIBLE_DEVICES="0,1,2,3" th main.lua \
    -data  ${data} \
    -train ${data}/list/train_gt.txt \
    -val ${data}/list/val_gt.txt \
    -dataset lane \
    -save experiments/models/${exp} \
    -retrain experiments/models/${exp}/ENet_concat_new.t7 \
    -shareGradInput true \
-   -nThreads 2 \
-   -nGPU 2 \
+   -nThreads 8 \
+   -nGPU 4 \
    -batchSize 12 \
    -maxIter 100000 \
    -LR 0.01 \