jimyoung6709 commited on
Commit
f510186
·
verified ·
1 Parent(s): a8a8070

Delete config.json

Browse files
Files changed (1) hide show
  1. 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
- }