-
Notifications
You must be signed in to change notification settings - Fork 20
/
run_training_one_site_out_docker.sh
38 lines (32 loc) · 1.38 KB
/
run_training_one_site_out_docker.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
#!/bin/bash
OAI_SRC=/media/lext/FAST/OA_progression_project/Data/
OAI_MOST_IMG_SRC=/media/lext/FAST/OA_progression_project/Data/MOST_OAI_00_0_2
WRKDIR=/media/lext/FAST/OA_progression_project/workdir
mkdir -p $WRKDIR/torch_models
docker build -t oaprog_img .
#nvidia-docker run -it --name oa_prog_training --rm \
# -v $WRKDIR:/workdir/:rw \
# -v $OAI_MOST_IMG_SRC:/data/:ro \
# -v $OAI_SRC:/oai_data_root:ro \
# -v $WRKDIR/torch_models:/root/.torch/models/ --ipc=host \
# oaprog_img python -u run_crossval_across_sites.py \
# --snapshots /workdir/snapshots \
# --logs /workdir/logs \
# --dataset_root /data/ \
# --metadata_root /workdir/Metadata \
# --oai_data_root /oai_data_root \
# --n_epochs 20
for SITE in A B C D E;
do
nvidia-docker run -it --name oa_prog_oof_inference --rm \
-v $WRKDIR:/workdir/:rw \
-v $OAI_MOST_IMG_SRC:/data/:ro \
-v $OAI_SRC:/oai_data_root:ro \
-v $WRKDIR/torch_models:/root/.torch/models/ --ipc=host \
oaprog_img python -u run_oof_inference.py \
--snapshots_root /workdir/snapshots/2019_09_15_12_18/ \
--snapshot site_${SITE}/ \
--save_dir /workdir/snapshots/2019_09_15_12_18/site_${SITE}/inference \
--dataset_root /data/ \
--metadata_root /workdir/Metadata
done;