-
Notifications
You must be signed in to change notification settings - Fork 9
/
ant-dir.json
36 lines (36 loc) · 897 Bytes
/
ant-dir.json
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
{
"env_name": "ant-dir",
"n_train_tasks": 2,
"n_eval_tasks": 2,
"latent_size": 20,
"env_params": {
"n_tasks": 2,
"forward_backward": 1
},
"algo_params": {
"meta_batch": 4,
"num_initial_steps": 2000,
"num_steps_prior": 400,
"num_steps_posterior": 0,
"num_extra_rl_steps_posterior": 600,
"num_train_steps_per_itr": 4000,
"num_evals": 1,
"num_steps_per_eval": 600,
"num_exp_traj_eval": 2,
"embedding_batch_size": 256,
"embedding_mini_batch_size": 256,
"data_dir": "",
"sample": 1,
"epoch": null,
"n_trj": 50,
"sparse_rewards": 0,
"alpha_init": 1e6,
"alpha_max": 5e6,
"train_epoch": null,
"eval_epoch": null,
"use_value_penalty": 1
},
"util_params": {
"use_gpu": 1
}
}