-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathtrainWithInception.sh
28 lines (20 loc) · 1019 Bytes
/
trainWithInception.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
25
26
27
28
#!/bin/bash
#SBATCH --job-name=tensorflowShowandTell${SLURM_ARRAY_TASK_ID}
#SBATCH --time=504:00:00
# To run as an array job, use the following command:
# sbatch --partition=beards --gres=gpu:1 --array=0-1 --mem=64000 --constraint=intel trainWithInception.sh
#TODO: Load your modules and your profile
module load cuda/8.0
source /home/thpaul/.bash_profile
#TODO: Update your Base Directory and Scripts directory:
WORK=/work/thpaul/tf_tools/tensorflow/im2txt
SCRIPTS="${WORK}/NLP_Project_Scripts"
#TODO: Directory containing preprocessed MSCOCO data.
MSCOCO_DIR="/work/thpaul/tf_tools/tensorflow/im2txt/im2txt/data/mscoco"
#TODO: Inception v3 checkpoint file.
INCEPTION_CHECKPOINT="/work/thpaul/tf_tools/tensorflow/im2txt/im2txt/data/inception_v3.ckpt"
#TODO: Directory to save the model.
MODEL_DIR="/work/thpaul/tf_tools/tensorflow/im2txt/im2txt/model"
# Build the model.
echo "Starting Task! trainWithInception${SLURM_ARRAY_TASK_ID}.sh from `pwd`"
(exec ${SCRIPTS}/trainWithInception${SLURM_ARRAY_TASK_ID}.sh)