Update setup.sh

setup.sh

@@ -26,8 +26,16 @@ echo
 echo "Next, pick the ${RESOURCE} partition on which you will be running your
 notebooks.  Default partition (gll_all_serial). To specify gpu partition use gll_usr_gpuprod  --gres=gpu:kepler:1"
 echo
-read -p "${RESOURCE} partition (default: gll_all_serial) > "  PARTITION
-PARTITION=${PARTITION:-gll_all_serial}
+read -p "${RESOURCE} partition (default: gll_usr_prod) > "  PARTITION
+PARTITION=${PARTITION:-gll_usr_prod}
+
+echo
+echo "Next, pick the ${RESOURCE} reservation which you will be using"
+echo
+read -p "${RESOURCE} partition (default: s_tra_eudat) > "  RESERVATION
+RESERVATION=${RESERVATION:-s_tra_eudat}
+
+PARTITION="${PARTITION} --reservation=${RESERVATION} --ntasks-per-node=18 --nodes=1"
 
 
 echo