-
Notifications
You must be signed in to change notification settings - Fork 0
/
run_point.sh
executable file
·92 lines (91 loc) · 2.93 KB
/
run_point.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
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
#!/bin/bash
# Script to reproduce results
Foldername="0723_offline_meta_rl_reccc"
mkdir out_logs/${Foldername} &> /dev/null
declare -a tasks=( "cpearl-sparse-point-robot" )
# "cpearl-sparse-point-robot"
declare -a algos=( "cpearl" )
##
declare -a seeds=( "26" )
declare -a datadirs=( "sparse-point-robot" )
# "sparse-point-robot-20"
declare -a is_sparses=( "0" )
declare -a use_bracs=( "0" )
declare -a use_information_bottlenecks=( "0" )
declare -a is_zlosses=( "1" )
declare -a is_onlineadapt_threses=( "0" )
declare -a is_onlineadapt_maxes=( "1" )
declare -a num_exp_traj_evals=( "10" )
declare -a allow_backward_zs=( "0" )
declare -a is_true_sparses=( "0" )
declare -a r_threses=( "-1000000" )
n=0
# 52 54 21
# 52 54 21 dense
gpunum=8
for task in "${tasks[@]}"
do
for algo in "${algos[@]}"
do
for seed in "${seeds[@]}"
do
for datadir in "${datadirs[@]}"
do
for is_sparse in "${is_sparses[@]}"
do
for use_brac in "${use_bracs[@]}"
do
for use_information_bottleneck in "${use_information_bottlenecks[@]}"
do
for is_zloss in "${is_zlosses[@]}"
do
for is_onlineadapt_thres in "${is_onlineadapt_threses[@]}"
do
for is_onlineadapt_max in "${is_onlineadapt_maxes[@]}"
do
for num_exp_traj_eval in "${num_exp_traj_evals[@]}"
do
for allow_backward_z in "${allow_backward_zs[@]}"
do
for is_true_sparse in "${is_true_sparses[@]}"
do
for r_thres in "${r_threses[@]}"
do
OMP_NUM_THREADS=16 KMP_AFFINITY="compact,granularity\=fine" nohup python launch_experiment_${algo}.py \
./configs/${task}.json \
./data/${datadir} \
--gpu=${n} \
--is_sparse_reward=${is_sparse} \
--use_brac=${use_brac} \
--use_information_bottleneck=${use_information_bottleneck} \
--is_zloss=${is_zloss} \
--is_onlineadapt_thres=${is_onlineadapt_thres} \
--is_onlineadapt_max=${is_onlineadapt_max} \
--num_exp_traj_eval=${num_exp_traj_eval} \
--allow_backward_z=${allow_backward_z} \
--is_true_sparse_rewards=${is_true_sparse} \
--r_thres=${r_thres} \
>& out_logs/${Foldername}/${task}_${algo}_${datadir}_${is_sparse}_${use_brac}_${use_information_bottleneck}_${is_zloss}_${is_onlineadapt_thres}_${is_onlineadapt_max}_${num_exp_traj_eval}_${allow_backward_z}_${is_true_sparse}_${r_thres}_${seed}_${n}.txt &
echo "task: ${task}, algo: ${algo}, datadir: ${datadir}, is_sparse: ${is_sparse}, use_brac: ${use_brac}"
echo " use_information_bottleneck: ${use_information_bottleneck}, is_zloss: ${is_zloss}"
echo " is_onlineadapt_thres: ${is_onlineadapt_thres}, is_onlineadapt_max: ${is_onlineadapt_max}"
echo " num_exp_traj_eval: ${num_exp_traj_eval}, allow_backward_z: ${allow_backward_z}"
echo " is_true_sparse: ${is_true_sparse}, r_thres: ${r_thres}, seed: ${seed}, GPU: $n"
n=$[($n+1) % ${gpunum}]
sleep 10
done
done
done
done
done
done
done
done
done
done
done
done
done
done
# CUDA_VISIBLE_DEVICES=3 python policy_train.py ./configs/sparse-point-robot.json
# python policy_train.py ./configs/sparse-point-robot.json --is_uniform