forked from ZhiboRao/NLCA_Net
-
Notifications
You must be signed in to change notification settings - Fork 0
/
TrainKitti2015.sh
executable file
·24 lines (24 loc) · 1.22 KB
/
TrainKitti2015.sh
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
#!/bin/bash
echo $"Starting Net..."
CUDA_VISIBLE_DEVICES=0,1 nohup python -u ./Source/main.py \
--gpu 2 --phase train \
--dataset KITTI \
--modelName CCANet \
--modelDir ./PAModel/ \
--auto_save_num 10 \
--imgNum 200 \
--valImgNum 0 \
--maxEpochs 800 \
--learningRate 0.001 \
--outputDir ./Result/ \
--trainListPath ./Dataset/trainlist_kitti_2015.txt \
--trainLabelListPath ./Dataset/labellist_kitti_2015.txt \
--valListPath ./Dataset/val_trainlist_kitti_2015.txt \
--valLabelListPath ./Dataset/val_labellist_kitti_2015.txt \
--corpedImgWidth 512 \
--corpedImgHeight 256 \
--batchSize 4 \
--pretrain false > TrainKitti.log 2>&1 &
echo $"You can get the running log via the command line that tail -f TrainKitti.log"
echo $"The result will be saved in the result folder"
echo $"If you have any questions, you could contact me. My email: [email protected]"