-
Notifications
You must be signed in to change notification settings - Fork 1
/
run.sh
104 lines (99 loc) · 2.99 KB
/
run.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
93
94
95
96
97
98
99
100
101
102
103
104
#############################
########## CRESP-T ##########
#############################
CUDA_VISIBLE_DEVICES=5 nohup python -u VRL/train.py \
task_name=walker_walk \
task=online_dmc \
num_train_frames=500001 \
batch_size=128 \
lr=0.001 \
critic_update_times=2 \
algo=crespt \
use_cf=True \
feature_dim=64 \
agent.rsd_nstep=2 \
agent.cf_temp=0.01 \
agent.cf_update_freq=1 \
agent.omega_num_sample=32 \
agent.omega_cf_output_mode=min_mu \
eval_on_dcs=True \
num_sources=3 \
dcs_type=background \
data_path=/home/amax/sda6/amax/rl/data/DAVIS/JPEGImages/480p/ \
use_tb=False \
seed=0 \
exp_name=crespt-rsd2-background3 > ./log/crespt/crespt-rsd2-background3-ww-s0.log 2>&1 &
# CUDA_VISIBLE_DEVICES=0 nohup python -u VRL/train.py \
# task_name=walker_walk \
# task=online_dmc \
# num_train_frames=500001 \
# batch_size=128 \
# lr=0.001 \
# critic_update_times=2 \
# algo=crespt \
# use_cf=True \
# feature_dim=64 \
# agent.rsd_nstep=2 \
# agent.cf_temp=0.01 \
# agent.cf_update_freq=1 \
# agent.omega_num_sample=32 \
# agent.omega_cf_output_mode=min_mu \
# eval_on_dcs=True \
# scale=0.1 \
# scale_end=0.2 \
# num_sources=3 \
# dcs_type=color \
# data_path=/home/amax/sda6/amax/rl/data/DAVIS/JPEGImages/480p/ \
# use_tb=False \
# seed=0 \
# exp_name=crespt-rsd2-color3_10_20 > ./log/crespt/crespt-rsd2-color3_10_20-ww-s0.log 2>&1 &
###########################
########## CRESP ##########
###########################
# CUDA_VISIBLE_DEVICES=3 nohup python -u VRL/train.py \
# task_name=walker_walk \
# task=online_dmc \
# num_train_frames=500001 \
# batch_size=128 \
# lr=0.001 \
# critic_update_times=2 \
# algo=cresp \
# use_cf=True \
# feature_dim=64 \
# agent.rsd_nstep=5 \
# agent.cf_temp=0.01 \
# agent.cf_update_freq=1 \
# agent.omega_num_sample=32 \
# agent.omega_cf_output_mode=min_mu \
# eval_on_dcs=True \
# num_sources=3 \
# dcs_type=background \
# data_path=/home/amax/sda6/amax/rl/data/DAVIS/JPEGImages/480p/ \
# use_tb=False \
# seed=0 \
# exp_name=cresp-rsd5-background3 > ./log/cresp/cresp-rsd5-background3-ww-s0.log 2>&1 &
# CUDA_VISIBLE_DEVICES=5 nohup python -u VRL/train.py \
# cuda_id=5 \
# task_name=walker_walk \
# task=online_dmc \
# num_train_frames=500001 \
# batch_size=128 \
# lr=0.001 \
# critic_update_times=2 \
# algo=cresp \
# use_cf=True \
# feature_dim=64 \
# agent.rsd_nstep=5 \
# agent.cf_temp=0.01 \
# agent.cf_update_freq=1 \
# agent.omega_num_sample=32 \
# agent.omega_cf_output_mode=min_mu \
# eval_on_dcs=True \
# scale=0.1 \
# scale_end=0.2 \
# num_sources=3 \
# dcs_type=color \
# data_path=/home/amax/sda6/amax/rl/data/DAVIS/JPEGImages/480p/ \
# use_tb=False \
# seed=0 \
# exp_name=cresp-rsd5-color2_10_20 > ./log/cresp/cresp-rsd5-color2_10_20-ww-s0.log 2>&1 &