forked from ryanxhr/DWBC
-
Notifications
You must be signed in to change notification settings - Fork 0
/
run_dwbc.sh
65 lines (49 loc) · 1.18 KB
/
run_dwbc.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
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
#!/bin/bash
# Script to reproduce results
GPU_LIST=(0 1 2 3)
for alpha in 7.5; do
for seed in 0 1 2 3 4; do
# setting 1
for env in "halfcheetah-expert-v2" "hopper-expert-v2" "walker2d-expert-v2" "ant-expert-v2"; do
for x in 30 60 90; do
GPU_DEVICE=${GPU_LIST[task%${#GPU_LIST[@]}]}
CUDA_VISIBLE_DEVICES=$GPU_DEVICE python main.py \
--algorithm "DWBC" \
--env $env \
--split_x $x \
--alpha $alpha \
--seed $seed &
sleep 2
let "task=$task+1"
done
done
# setting 2
for env in "halfcheetah-medium-replay-v2" "hopper-medium-replay-v2" "walker2d-medium-replay-v2" "ant-medium-replay-v2"; do
for x in 2 5 10; do
GPU_DEVICE=${GPU_LIST[task%${#GPU_LIST[@]}]}
CUDA_VISIBLE_DEVICES=$GPU_DEVICE python main.py \
--algorithm "DWBC" \
--env $env \
--split_x $x \
--alpha $alpha \
--seed $seed &
sleep 2
let "task=$task+1"
done
done
# setting 3
for env in "pen-expert-v1" "door-expert-v1" "hammer-expert-v1" "relocate-expert-v1"; do
for x in 30 60 90; do
GPU_DEVICE=${GPU_LIST[task%${#GPU_LIST[@]}]}
CUDA_VISIBLE_DEVICES=$GPU_DEVICE python main.py \
--algorithm "DWBC" \
--env $env \
--split_x $x \
--alpha $alpha \
--seed $seed &
sleep 2
let "task=$task+1"
done
done
done
done