Minor tweaks to script files

code/train-model.sh

@@ -2,7 +2,7 @@
 #
 #SBATCH --partition=longq    # Partition to submit to
 #
-#SBATCH --time=0-08:00         # Runtime in D-HH:MM
+#SBATCH --time=3-00:00         # Runtime in D-HH:MM
 #SBATCH --nodes=1
 #SBATCH --ntasks=1
 #SBATCH --ntasks-per-node=1
@@ -14,8 +14,9 @@ RUN_ID=$1
 MODEL_NAME=$2
 NUM_EPOCHS=$3
 NUM_USERS=$4
+TIME=$5
 
 source ~/mypython/bin/activate
 python main.py $RUN_ID $MODEL_NAME $NUM_EPOCHS $NUM_USERS
 
-
+# mailx -S smtp=mail.cs.umass.edu -a ../logs/usc-isi/${TIME}/train.log -a ../logs/usc-isi/${TIME}/train.err -s "Test mail" [email protected] < /dev/null

code/training-job.sh

@@ -46,6 +46,9 @@ TIME=`(date +%Y-%m-%d-%H-%M-%S)`
                 -e ../logs/usc-isi/${TIME}/train.err \
                 -o ../logs/usc-isi/${TIME}/train.log \
                 --cpus-per-task $NUM_THREADS \
-                train-model.sh $RUN_ID $MODEL_NAME $NUM_EPOCHS $NUM_USERS
+                --mail-type=ALL \
+                [email protected],[email protected] \
+                train-model.sh $RUN_ID $MODEL_NAME $NUM_EPOCHS $NUM_USERS $TIME
+
 exit