-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathil_eval.yaml
56 lines (54 loc) · 1.23 KB
/
il_eval.yaml
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
program: imitate_episodes.py
method: grid
parameters:
task_name:
# value: "sim_insertion_scripted"
value: "sim_transfer_cube_scripted"
policy_class:
value: "ACT"
kl_weight:
value: 10
decay_rate:
values: [1]
chunk_size:
values: [100]
ckpt_dir:
# value: "/localdata/yy/zzzzworkspace/act/ckpt/sim_transfer_cube_scripted_run3"
# value: "./ckpt/sim_insertion_scripted/Testchunk"
# value: "/localdata/yy/zzzzworkspace/act/ckpt/sim_insertion_human_chunk100"
# value: "/localdata/yy/zzzzworkspace/act/ckpt/sim_transfer_cube_scripted_run2"
value: "/home/torchuser/adaptiveAC/wandb_ckpt/vae_transfer_cube"
prediction_ckpt_dir:
value: "/home/torchuser/adaptiveAC/wandb_ckpt/vae_transfer_cube_prediction"
hidden_dim:
value: 512
batch_size:
value: 8
dim_feedforward:
value: 3200
query_freq:
values: [100]
temporal_agg:
value: false
num_epochs:
value: 2000
num_epochs_prediction:
value: 5
lr:
value: 1e-5
seed:
values: [1, 2, 3, 4, 5]
eval:
value: true
onscreen_render:
value: false
use_predict_model:
value: false
state_dim:
value: 14
action_dim:
value: 14
command:
- python3
- ${program}
- ${args_no_boolean_flags}