-
Notifications
You must be signed in to change notification settings - Fork 0
/
run_models_in_sequence.sh
executable file
·45 lines (38 loc) · 1.47 KB
/
run_models_in_sequence.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
#!/usr/bin/bash
# this env variable is used to change the dockerbuild output directory
export TMPDIR=/data/dssg/occrp/data/docker
name1="EfficientNetB0"
name2="EfficientNetB0BW"
name3="EfficientNetB4"
name4="EfficientNetB4BW"
podman run -d \
--security-opt=label=disable \
-v /data/dssg/occrp/data:/data \
--hooks-dir=/usr/share/containers/oci/hooks.d/ \
--cap-add SYS_ADMIN \
--name $name1 \
$(podman build -f gpu.Dockerfile -t train-gpu -q) train-feature-extraction --model-name EfficientNetB4
podman run -d \
--security-opt=label=disable \
-v /data/dssg/occrp/data:/data \
--hooks-dir=/usr/share/containers/oci/hooks.d/ \
--cap-add SYS_ADMIN \
--name $name2 \
$(podman build -f train_gpu.Dockerfile -t train-gpu -q) train-feature-extraction --model-name EfficientNetB0BW
podman wait --name=$name2 \
podman run -d \
--security-opt=label=disable \
-v /data/dssg/occrp/data:/data \
--hooks-dir=/usr/share/containers/oci/hooks.d/ \
--cap-add SYS_ADMIN \
--name $name3 \
$(podman build -f train_gpu.Dockerfile -t train-gpu -q) train-feature-extraction --model-name EfficientNetB4
podman wait --name=$name3 \
podman run -d \
--security-opt=label=disable \
-v /data/dssg/occrp/data:/data \
--hooks-dir=/usr/share/containers/oci/hooks.d/ \
--cap-add SYS_ADMIN \
--name $name4 \
$(podman build -f gpu.Dockerfile -t train-gpu -q) train-feature-extraction --model-name EfficientNetB4BW
podman wait $name4