Sync

_ShanghaiA-train.sh

@@ -6,6 +6,8 @@
 #SBATCH --mem=24000
 #SBATCH --time=3-00:00:00
 
+set -e
+
 export CUDA_HOME=/opt/cuda-9.0.176.1/
 export CUDNN_HOME=/opt/cuDNN-7.0/
 export STUDENT_ID=$(whoami)
@@ -21,8 +23,11 @@ export TMPDIR=/disk/scratch/${STUDENT_ID}/
 export TMP=/disk/scratch/${STUDENT_ID}/
 
 source /home/${STUDENT_ID}/miniconda3/bin/activate mlp-cuda
+
 python train.py \
-    --model='stn' \
+    --debug True \
+    --model 'stn' \
     --save_path ./save_file/ShanghaiA \
     --batch_size 4 \
-    --gpus 0,1,2,3,4,5
+    --gpus 0,1,2,3,4,5 \
+    --print_freq 100