-
Notifications
You must be signed in to change notification settings - Fork 1
/
Copy pathpitchnet_evaluate_best_job.sh
46 lines (43 loc) · 1.48 KB
/
pitchnet_evaluate_best_job.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
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
#!/bin/bash
#
#SBATCH --job-name=pitchnet_eval
#SBATCH --out="/om2/user/msaddler/pitchnet/slurm_pitchnet_eval-%A_%a.out"
##SBATCH --gres=gpu:tesla-v100:1
##SBATCH --gres=gpu:QUADRORTX6000:1
#SBATCH --gres=gpu:GEFORCEGTX1080TI:1
##SBATCH --gres=gpu:GEFORCERTX2080TI:1
##SBATCH --gres=gpu:1
##SBATCH --constraint=high-capacity
#SBATCH --mem=12000
#SBATCH --cpus-per-task=6
#SBATCH --time=0-12:00:00
##SBATCH --exclude=node063
##SBATCH --partition=mcdermott
##SBATCH --partition=use-everything
##SBATCH --array=0-9
#SBATCH --array=83,154,190,191,286,288,302,335,338,346
PATH_DATA="/om/user/msaddler/data_pitchnet"
# PATH_DATA="$SCRATCH_PATH/data_pitchnet"
PATH_SAVED_MODELS="/om2/user/msaddler/pitchnet/saved_models"
PATH_PACKAGES="/om2/user/msaddler/pitchnet/packages"
ZPJID=$(printf "%04d" $SLURM_ARRAY_TASK_ID)
OUTDIR='/saved_models/arch_search_v02_topN/DEMO/arch_'$ZPJID
DATA_TAG="sr20000_cf100_species002_spont070_BW10eN1_IHC3000Hz_IHC7order"
# TFRECORDS_REGEX="stim_waveform.tfrecords"
TFRECORDS_REGEX="$DATA_TAG/bez2018meanrates_*.tfrecords"
EFN_PREFIX='EVAL_SOFTMAX_'
OUTPUT_LOG_FN=$OUTDIR'/output_eval.log'
echo "[START EVALUATION] $OUTPUT_LOG_FN"
singularity exec --nv \
-B /home \
-B /nobackup \
-B /om \
-B /om2 \
-B /om4 \
-B /nese \
-B $PATH_PACKAGES:/packages \
-B $PATH_DATA:/data \
-B $PATH_SAVED_MODELS:/saved_models \
/om2/user/msaddler/vagrant/tensorflow-1.13.1-pitchnet.simg \
./pitchnet_run_eval.sh $OUTDIR $TFRECORDS_REGEX $EFN_PREFIX $OUTPUT_LOG_FN
echo "[END EVALUATION] $OUTPUT_LOG_FN"