-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathrun_experiments_bottleneck.sh
42 lines (32 loc) · 1.89 KB
/
run_experiments_bottleneck.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
CONFIG_DIR="./configs_experiments/bottleneck_base_grid_depths_layers/"
mkdir logs
for CONFIG in "$CONFIG_DIR"bottleneck_base-name=depth_2-*.yaml; do
if [ "$CONFIG" != "$CONFIG_DIR/*.yaml" ]; then
sbatch --gpus=1 --wrap="python main.py --cfg $CONFIG --repeat 1 --mark_done" --time=24:00:00 --ntasks-per-node=1 --mem-per-cpu=2GB --cpus-per-task=16 --output=logs/%j.out
fi
done
for CONFIG in "$CONFIG_DIR"bottleneck_base-name=depth_3-*.yaml; do
if [ "$CONFIG" != "$CONFIG_DIR/*.yaml" ]; then
sbatch --gpus=1 --wrap="python main.py --cfg $CONFIG --repeat 1 --mark_done" --time=24:00:00 --ntasks-per-node=1 --mem-per-cpu=2GB --cpus-per-task=16 --output=logs/%j.out
fi
done
for CONFIG in "$CONFIG_DIR"bottleneck_base-name=depth_4-*.yaml; do
if [ "$CONFIG" != "$CONFIG_DIR/*.yaml" ]; then
sbatch --gpus=1 --wrap="python main.py --cfg $CONFIG --repeat 1 --mark_done" --time=24:00:00 --ntasks-per-node=1 --mem-per-cpu=2GB --cpus-per-task=16 --output=logs/%j.out
fi
done
for CONFIG in "$CONFIG_DIR"bottleneck_base-name=depth_5-*.yaml; do
if [ "$CONFIG" != "$CONFIG_DIR/*.yaml" ]; then
sbatch --gpus=1 --wrap="python main.py --cfg $CONFIG --repeat 1 --mark_done" --time=24:00:00 --ntasks-per-node=1 --mem-per-cpu=2GB --cpus-per-task=16 --output=logs/%j.out
fi
done
for CONFIG in "$CONFIG_DIR"bottleneck_base-name=depth_6-*.yaml; do
if [ "$CONFIG" != "$CONFIG_DIR/*.yaml" ]; then
sbatch --gpus=1 --gres=gpumem:20g --wrap="python main.py --cfg $CONFIG --repeat 1 --mark_done" --time=24:00:00 --ntasks-per-node=1 --mem-per-cpu=16GB --cpus-per-task=16 --output=logs/%j.out
fi
done
for CONFIG in "$CONFIG_DIR"bottleneck_base-name=depth_7-*.yaml; do
if [ "$CONFIG" != "$CONFIG_DIR/*.yaml" ]; then
sbatch --gpus=1 --gres=gpumem:30g --wrap="python main.py --cfg $CONFIG --repeat 1 --mark_done" --time=24:00:00 --ntasks-per-node=1 --mem-per-cpu=16GB --cpus-per-task=16 --output=logs/%j.out
fi
done