Delete config.json
Browse files- config.json +0 -255
config.json
DELETED
@@ -1,255 +0,0 @@
|
|
1 |
-
{
|
2 |
-
"algo_name": "bc",
|
3 |
-
"experiment": {
|
4 |
-
"name": "drp_robomimic",
|
5 |
-
"validate": false,
|
6 |
-
"logging": {
|
7 |
-
"terminal_output_to_txt": false,
|
8 |
-
"log_tb": true,
|
9 |
-
"log_wandb": true,
|
10 |
-
"wandb_proj_name": "neural_mp"
|
11 |
-
},
|
12 |
-
"save": {
|
13 |
-
"enabled": true,
|
14 |
-
"every_n_seconds": null,
|
15 |
-
"every_n_epochs": 500,
|
16 |
-
"epochs": [],
|
17 |
-
"on_best_validation": false,
|
18 |
-
"on_best_rollout_return": false,
|
19 |
-
"on_best_rollout_success_rate": true
|
20 |
-
},
|
21 |
-
"epoch_every_n_steps": 1000,
|
22 |
-
"validation_epoch_every_n_steps": 1000,
|
23 |
-
"env": null,
|
24 |
-
"additional_envs": null,
|
25 |
-
"render": false,
|
26 |
-
"render_video": true,
|
27 |
-
"keep_all_videos": true,
|
28 |
-
"video_skip": 5,
|
29 |
-
"rollout": {
|
30 |
-
"enabled": true,
|
31 |
-
"n": 25,
|
32 |
-
"horizon": 150,
|
33 |
-
"rate": 500,
|
34 |
-
"warmstart": 0,
|
35 |
-
"terminate_on_success": true,
|
36 |
-
"evaluate_on_dataset": false,
|
37 |
-
"evaluate_on_dataset_filter_key": "valid"
|
38 |
-
},
|
39 |
-
"pcd_params": {
|
40 |
-
"num_robot_points": 2048,
|
41 |
-
"num_obstacle_points": 4096,
|
42 |
-
"target_pcd_type": "joint",
|
43 |
-
"noise_scale": 0.0,
|
44 |
-
"normalize_joint_angles": false,
|
45 |
-
"relabel_goal_angles": false
|
46 |
-
},
|
47 |
-
"num_envs": 20
|
48 |
-
},
|
49 |
-
"train": {
|
50 |
-
"pad_same": true,
|
51 |
-
"data": "/home/jiahuiya/Neural_MP_Proj/neural_mp/datasets/v1_400k.hdf5",
|
52 |
-
"output_dir": "logs/mp/trained_models",
|
53 |
-
"num_data_workers": 16,
|
54 |
-
"hdf5_cache_mode": null,
|
55 |
-
"hdf5_use_swmr": true,
|
56 |
-
"hdf5_load_next_obs": false,
|
57 |
-
"hdf5_normalize_obs": false,
|
58 |
-
"hdf5_filter_key": "train",
|
59 |
-
"hdf5_validation_filter_key": null,
|
60 |
-
"seq_length": 2,
|
61 |
-
"pad_seq_length": true,
|
62 |
-
"frame_stack": 1,
|
63 |
-
"pad_frame_stack": true,
|
64 |
-
"dataset_keys": [
|
65 |
-
"actions"
|
66 |
-
],
|
67 |
-
"goal_mode": null,
|
68 |
-
"cuda": true,
|
69 |
-
"batch_size": 16,
|
70 |
-
"num_epochs": 10000,
|
71 |
-
"seed": 1
|
72 |
-
},
|
73 |
-
"algo": {
|
74 |
-
"optim_params": {
|
75 |
-
"policy": {
|
76 |
-
"optimizer_type": "adam",
|
77 |
-
"learning_rate": {
|
78 |
-
"initial": 0.0001,
|
79 |
-
"decay_factor": 0.1,
|
80 |
-
"epoch_schedule": [],
|
81 |
-
"scheduler_type": "multistep"
|
82 |
-
},
|
83 |
-
"regularization": {
|
84 |
-
"L2": 0.0
|
85 |
-
}
|
86 |
-
}
|
87 |
-
},
|
88 |
-
"loss": {
|
89 |
-
"l2_weight": 1.0,
|
90 |
-
"l1_weight": 0.0,
|
91 |
-
"cos_weight": 0.0
|
92 |
-
},
|
93 |
-
"actor_layer_dims": [],
|
94 |
-
"actor_activation": "nn.ReLU",
|
95 |
-
"gaussian": {
|
96 |
-
"enabled": false,
|
97 |
-
"fixed_std": false,
|
98 |
-
"init_std": 0.1,
|
99 |
-
"min_std": 0.01,
|
100 |
-
"std_activation": "softplus",
|
101 |
-
"low_noise_eval": true
|
102 |
-
},
|
103 |
-
"gmm": {
|
104 |
-
"enabled": true,
|
105 |
-
"num_modes": 5,
|
106 |
-
"min_std": 0.0001,
|
107 |
-
"std_activation": "softplus",
|
108 |
-
"low_noise_eval": true
|
109 |
-
},
|
110 |
-
"vae": {
|
111 |
-
"enabled": false,
|
112 |
-
"latent_dim": 14,
|
113 |
-
"latent_clip": null,
|
114 |
-
"kl_weight": 1.0,
|
115 |
-
"decoder": {
|
116 |
-
"is_conditioned": true,
|
117 |
-
"reconstruction_sum_across_elements": false
|
118 |
-
},
|
119 |
-
"prior": {
|
120 |
-
"learn": false,
|
121 |
-
"is_conditioned": false,
|
122 |
-
"use_gmm": false,
|
123 |
-
"gmm_num_modes": 10,
|
124 |
-
"gmm_learn_weights": false,
|
125 |
-
"use_categorical": false,
|
126 |
-
"categorical_dim": 10,
|
127 |
-
"categorical_gumbel_softmax_hard": false,
|
128 |
-
"categorical_init_temp": 1.0,
|
129 |
-
"categorical_temp_anneal_step": 0.001,
|
130 |
-
"categorical_min_temp": 0.3
|
131 |
-
},
|
132 |
-
"encoder_layer_dims": [
|
133 |
-
300,
|
134 |
-
400
|
135 |
-
],
|
136 |
-
"decoder_layer_dims": [
|
137 |
-
300,
|
138 |
-
400
|
139 |
-
],
|
140 |
-
"prior_layer_dims": [
|
141 |
-
300,
|
142 |
-
400
|
143 |
-
]
|
144 |
-
},
|
145 |
-
"rnn": {
|
146 |
-
"enabled": true,
|
147 |
-
"horizon": 2,
|
148 |
-
"hidden_dim": 1024,
|
149 |
-
"rnn_type": "LSTM",
|
150 |
-
"num_layers": 2,
|
151 |
-
"open_loop": false,
|
152 |
-
"kwargs": {
|
153 |
-
"bidirectional": false
|
154 |
-
}
|
155 |
-
},
|
156 |
-
"transformer": {
|
157 |
-
"enabled": false,
|
158 |
-
"context_length": 10,
|
159 |
-
"embed_dim": 512,
|
160 |
-
"num_layers": 6,
|
161 |
-
"num_heads": 8,
|
162 |
-
"emb_dropout": 0.1,
|
163 |
-
"attn_dropout": 0.1,
|
164 |
-
"block_output_dropout": 0.1,
|
165 |
-
"sinusoidal_embedding": false,
|
166 |
-
"activation": "gelu",
|
167 |
-
"supervise_all_steps": false,
|
168 |
-
"nn_parameter_for_timesteps": true
|
169 |
-
}
|
170 |
-
},
|
171 |
-
"observation": {
|
172 |
-
"modalities": {
|
173 |
-
"obs": {
|
174 |
-
"low_dim": [
|
175 |
-
"current_angles",
|
176 |
-
"goal_angles"
|
177 |
-
],
|
178 |
-
"rgb": [],
|
179 |
-
"depth": [],
|
180 |
-
"scan": [],
|
181 |
-
"pcd": [
|
182 |
-
"compute_pcd_params"
|
183 |
-
]
|
184 |
-
},
|
185 |
-
"goal": {
|
186 |
-
"low_dim": [],
|
187 |
-
"rgb": [],
|
188 |
-
"depth": [],
|
189 |
-
"scan": [],
|
190 |
-
"pcd": []
|
191 |
-
}
|
192 |
-
},
|
193 |
-
"encoder": {
|
194 |
-
"low_dim": {
|
195 |
-
"core_class": null,
|
196 |
-
"core_kwargs": {},
|
197 |
-
"obs_randomizer_class": null,
|
198 |
-
"obs_randomizer_kwargs": {}
|
199 |
-
},
|
200 |
-
"rgb": {
|
201 |
-
"core_class": "VisualCore",
|
202 |
-
"core_kwargs": {
|
203 |
-
"feature_dimension": 64,
|
204 |
-
"backbone_class": "ResNet18Conv",
|
205 |
-
"backbone_kwargs": {
|
206 |
-
"pretrained": false,
|
207 |
-
"input_coord_conv": false
|
208 |
-
},
|
209 |
-
"pool_class": "SpatialSoftmax",
|
210 |
-
"pool_kwargs": {
|
211 |
-
"num_kp": 32,
|
212 |
-
"learnable_temperature": false,
|
213 |
-
"temperature": 1.0,
|
214 |
-
"noise_std": 0.0
|
215 |
-
}
|
216 |
-
},
|
217 |
-
"obs_randomizer_class": "CropRandomizer",
|
218 |
-
"obs_randomizer_kwargs": {
|
219 |
-
"crop_height": 224,
|
220 |
-
"crop_width": 224,
|
221 |
-
"num_crops": 1,
|
222 |
-
"pos_enc": false
|
223 |
-
}
|
224 |
-
},
|
225 |
-
"depth": {
|
226 |
-
"core_class": "VisualCore",
|
227 |
-
"core_kwargs": {},
|
228 |
-
"obs_randomizer_class": null,
|
229 |
-
"obs_randomizer_kwargs": {}
|
230 |
-
},
|
231 |
-
"scan": {
|
232 |
-
"core_class": "ScanCore",
|
233 |
-
"core_kwargs": {},
|
234 |
-
"obs_randomizer_class": null,
|
235 |
-
"obs_randomizer_kwargs": {}
|
236 |
-
},
|
237 |
-
"pcd": {
|
238 |
-
"core_class": "PcdCore",
|
239 |
-
"core_kwargs": {
|
240 |
-
"backbone_class": "PointNetEncoder",
|
241 |
-
"backbone_kwargs": {
|
242 |
-
"encoder_size": "small"
|
243 |
-
}
|
244 |
-
},
|
245 |
-
"obs_randomizer_class": null,
|
246 |
-
"obs_randomizer_kwargs": {}
|
247 |
-
}
|
248 |
-
}
|
249 |
-
},
|
250 |
-
"meta": {
|
251 |
-
"hp_base_config_file": null,
|
252 |
-
"hp_keys": [],
|
253 |
-
"hp_values": []
|
254 |
-
}
|
255 |
-
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|