Hugging Face's logo

mihdalal
/
NeuralMP

Robotics
PyTorch
English
motion planning
Model card Files Community
mihdalal commited on
Commit
198a575
•
1 Parent(s): 55f1265

Upload 2 files

Browse files
Files changed (2) hide show
  1. config.json +294 -0
  2. pytorch_model.bin +3 -0
config.json ADDED
@@ -0,0 +1,294 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "algo_name": "bc",
3
+ "config": {
4
+ "algo_name": "bc",
5
+ "experiment": {
6
+ "name": "table_10M_final_smooth_cubic_with_objaverse",
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": 250,
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": 1,
31
+ "rollout": {
32
+ "enabled": false,
33
+ "n": 10,
34
+ "horizon": 300,
35
+ "rate": 250,
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
+ "dagger": {
51
+ "enabled": false,
52
+ "online_epoch_rate": 50,
53
+ "num_episodes": 10,
54
+ "resampling_strategy": "all",
55
+ "num_steps_to_keep_before_collision": 10,
56
+ "data_mode": "all",
57
+ "dagger_traj_filter": "all",
58
+ "num_trajs_to_relabel": 1
59
+ }
60
+ },
61
+ "train": {
62
+ "pad_same": true,
63
+ "data": "../neural_mp/datasets/final_datasets_v2/table_100K_single_layer_final_smooth_cubic_no_objaverse.hdf5",
64
+ "output_dir": "logs/mp/trained_models",
65
+ "num_data_workers": 16,
66
+ "hdf5_cache_mode": null,
67
+ "hdf5_use_swmr": true,
68
+ "hdf5_load_next_obs": false,
69
+ "hdf5_normalize_obs": false,
70
+ "hdf5_filter_key": "train",
71
+ "hdf5_validation_filter_key": null,
72
+ "seq_length": 2,
73
+ "pad_seq_length": true,
74
+ "frame_stack": 1,
75
+ "pad_frame_stack": true,
76
+ "dataset_keys": [
77
+ "actions"
78
+ ],
79
+ "goal_mode": null,
80
+ "cuda": true,
81
+ "batch_size": 16,
82
+ "num_epochs": 10000000,
83
+ "seed": 1
84
+ },
85
+ "algo": {
86
+ "optim_params": {
87
+ "policy": {
88
+ "optimizer_type": "adam",
89
+ "learning_rate": {
90
+ "initial": 0.0001,
91
+ "decay_factor": 0.1,
92
+ "epoch_schedule": [],
93
+ "scheduler_type": "multistep"
94
+ },
95
+ "regularization": {
96
+ "L2": 0.0
97
+ }
98
+ }
99
+ },
100
+ "loss": {
101
+ "l2_weight": 1.0,
102
+ "l1_weight": 0.0,
103
+ "cos_weight": 0.0,
104
+ "collision_weight": 0.0,
105
+ "point_match_weight": 0.0,
106
+ "exponential_precision_weight": 0.0,
107
+ "collision_loss_params": {
108
+ "margin": 0.03,
109
+ "hinge_loss": true,
110
+ "reduction": "mean",
111
+ "smooth_sdf_loss": "none",
112
+ "compute_loss_on_penetrations_only": false
113
+ }
114
+ },
115
+ "actor_layer_dims": [],
116
+ "actor_activation": "nn.ReLU",
117
+ "gaussian": {
118
+ "enabled": false,
119
+ "fixed_std": false,
120
+ "init_std": 0.1,
121
+ "min_std": 0.01,
122
+ "std_activation": "softplus",
123
+ "low_noise_eval": true
124
+ },
125
+ "gmm": {
126
+ "enabled": true,
127
+ "num_modes": 5,
128
+ "min_std": 0.0001,
129
+ "std_activation": "softplus",
130
+ "low_noise_eval": true
131
+ },
132
+ "vae": {
133
+ "enabled": false,
134
+ "latent_dim": 14,
135
+ "latent_clip": null,
136
+ "kl_weight": 1.0,
137
+ "decoder": {
138
+ "is_conditioned": true,
139
+ "reconstruction_sum_across_elements": false
140
+ },
141
+ "prior": {
142
+ "learn": false,
143
+ "is_conditioned": false,
144
+ "use_gmm": false,
145
+ "gmm_num_modes": 10,
146
+ "gmm_learn_weights": false,
147
+ "use_categorical": false,
148
+ "categorical_dim": 10,
149
+ "categorical_gumbel_softmax_hard": false,
150
+ "categorical_init_temp": 1.0,
151
+ "categorical_temp_anneal_step": 0.001,
152
+ "categorical_min_temp": 0.3
153
+ },
154
+ "encoder_layer_dims": [
155
+ 300,
156
+ 400
157
+ ],
158
+ "decoder_layer_dims": [
159
+ 300,
160
+ 400
161
+ ],
162
+ "prior_layer_dims": [
163
+ 300,
164
+ 400
165
+ ]
166
+ },
167
+ "rnn": {
168
+ "enabled": true,
169
+ "horizon": 2,
170
+ "hidden_dim": 1024,
171
+ "rnn_type": "LSTM",
172
+ "num_layers": 2,
173
+ "open_loop": false,
174
+ "kwargs": {
175
+ "bidirectional": false
176
+ }
177
+ },
178
+ "transformer": {
179
+ "enabled": false,
180
+ "context_length": 10,
181
+ "embed_dim": 512,
182
+ "num_layers": 6,
183
+ "num_heads": 8,
184
+ "emb_dropout": 0.1,
185
+ "attn_dropout": 0.1,
186
+ "block_output_dropout": 0.1,
187
+ "sinusoidal_embedding": false,
188
+ "activation": "gelu",
189
+ "supervise_all_steps": false,
190
+ "nn_parameter_for_timesteps": true,
191
+ "act_enabled": false
192
+ },
193
+ "mpinets": {
194
+ "enabled": false
195
+ }
196
+ },
197
+ "observation": {
198
+ "modalities": {
199
+ "obs": {
200
+ "low_dim": [
201
+ "current_angles",
202
+ "goal_angles"
203
+ ],
204
+ "rgb": [],
205
+ "depth": [],
206
+ "scan": [],
207
+ "pcd": [
208
+ "compute_pcd_params"
209
+ ]
210
+ },
211
+ "goal": {
212
+ "low_dim": [],
213
+ "rgb": [],
214
+ "depth": [],
215
+ "scan": [],
216
+ "pcd": []
217
+ }
218
+ },
219
+ "encoder": {
220
+ "low_dim": {
221
+ "core_class": null,
222
+ "core_kwargs": {},
223
+ "obs_randomizer_class": null,
224
+ "obs_randomizer_kwargs": {}
225
+ },
226
+ "rgb": {
227
+ "core_class": "VisualCore",
228
+ "core_kwargs": {
229
+ "feature_dimension": 64,
230
+ "backbone_class": "ResNet18Conv",
231
+ "backbone_kwargs": {
232
+ "pretrained": false,
233
+ "input_coord_conv": false
234
+ },
235
+ "pool_class": "SpatialSoftmax",
236
+ "pool_kwargs": {
237
+ "num_kp": 32,
238
+ "learnable_temperature": false,
239
+ "temperature": 1.0,
240
+ "noise_std": 0.0
241
+ }
242
+ },
243
+ "obs_randomizer_class": "CropRandomizer",
244
+ "obs_randomizer_kwargs": {
245
+ "crop_height": 224,
246
+ "crop_width": 224,
247
+ "num_crops": 1,
248
+ "pos_enc": false
249
+ }
250
+ },
251
+ "depth": {
252
+ "core_class": "VisualCore",
253
+ "core_kwargs": {},
254
+ "obs_randomizer_class": null,
255
+ "obs_randomizer_kwargs": {}
256
+ },
257
+ "scan": {
258
+ "core_class": "ScanCore",
259
+ "core_kwargs": {},
260
+ "obs_randomizer_class": null,
261
+ "obs_randomizer_kwargs": {}
262
+ },
263
+ "pcd": {
264
+ "core_class": "PcdCore",
265
+ "core_kwargs": {
266
+ "backbone_class": "PointNetEncoder",
267
+ "backbone_kwargs": {
268
+ "encoder_size": "small"
269
+ }
270
+ },
271
+ "obs_randomizer_class": null,
272
+ "obs_randomizer_kwargs": {}
273
+ }
274
+ }
275
+ },
276
+ "meta": {
277
+ "hp_base_config_file": null,
278
+ "hp_keys": [],
279
+ "hp_values": []
280
+ }
281
+ },
282
+ "obs_key_shapes": {
283
+ "compute_pcd_params": [
284
+ 673
285
+ ],
286
+ "current_angles": [
287
+ 7
288
+ ],
289
+ "goal_angles": [
290
+ 7
291
+ ]
292
+ },
293
+ "ac_dim": 7
294
+ }
pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6c07d2e51112e21e1479e4186df3e8ce0e73839750b372b99b4c80bfdf7992fb
3
+ size 86410961