Change data path

Author: ChienHsuan

configs/test.yaml

@@ -2,7 +2,7 @@ use_gpu: True
 
 model:
   name: 'osnet_ain_x0_5'
-  load_weights: '/home/lab314/HDD/Hsuan/models/osnet_ain_x0_5/model_best.pth.tar'
+  load_weights: '/USER-DEFINED-PATH/model_best.pth.tar'
   pretrained: False
 
 data:

test.sh

@@ -1,8 +1,8 @@
-DATA_DIR="/home/lab314/HDD/Dataset"
+DATA_DIR="/USER-DEFINED-PATH/Dataset/"
 CONFIG_FILE="configs/test.yaml"
 
 echo "Begin to test."
-export CUDA_VISIBLE_DEVICES=4
+export CUDA_VISIBLE_DEVICES=0
 python scripts/main.py \
 	--config-file ${CONFIG_FILE} \
 	--root ${DATA_DIR} \

train.sh

@@ -1,4 +1,4 @@
-DATA_DIR="/nfs/datasets/"
+DATA_DIR="/USER-DEFINED-PATH/Dataset/"
 CONFIG_FILE="configs/osnet_ain_x0_5.yaml"
 
 echo "Begin to train."