-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathrun_training_experiments
More file actions
executable file
·33 lines (25 loc) · 1.04 KB
/
run_training_experiments
File metadata and controls
executable file
·33 lines (25 loc) · 1.04 KB
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
#!/usr/bin/env bash
set -eu
IMG='docker.io/deepprojects/camelyon-cnns:tests'
OPTS='-u 1000:1000 --runtime=nvidia --device /dev/fuse --cap-add SYS_ADMIN --security-opt apparmor:unconfined --workdir=/home/cc/out'
# more information on merged.hf5: https://github.com/deep-projects/camelyon-experiments-v2
CMD='camelyon-cnns /home/cc/input/merged.hdf5'
# INPDIR_TYPE and INPDIR are set to run performance tests on a remote NFS share
# Change INPDIR_TYPE to SSD and INPDIR to an appropriate data location on your local SSD
INPDIR_TYPE="NFS"
#INPDIR_TYPE="SSD"
INPDIR="/remote_nfs/inputs"
#INPDIR="/local_ssd/inputs"
OUTDIR="/outputs"
docker pull ${IMG}
mkdir ${OUTDIR}
chmod 777 ${OUTDIR}
outdir="${OUTDIR}/${INPDIR_TYPE}"
mkdir ${outdir}
chmod 777 ${outdir}
for i in $(seq 1 9); do
outdir_n="${outdir}/${i}"
mkdir ${outdir_n}
chmod 777 ${outdir_n}
docker run ${OPTS} -e NVIDIA_VISIBLE_DEVICES=0 -v ${INPDIR}:/home/cc/input -v ${outdir_n}:/home/cc/out:rw ${IMG} ${CMD} --arch 1 --fast-hdf5 1 --mask-threshold 0.1 --epochs 8 --queue-size 1
done