Hugging Face's logo

jimyoung6709
/
DRP

Robotics
PyTorch
English
motion planning
Model card Files
xet
Community
jimyoung6709 commited on
Commit
506ff65
·
verified ·
1 Parent(s): f510186

Upload config.json

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