serialexperimentsleon commited on
Commit
37d986b
·
verified ·
1 Parent(s): b7a3e31

Add files using upload-large-folder tool

Browse files
config.yaml ADDED
@@ -0,0 +1,583 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ root_dir: /mnt/kostas-graid/datasets/extrinsic_contact_data
2
+ replay_buffer_size: 150000
3
+ replay_buffer_num_workers: 2
4
+ nstep: 3
5
+ batch_size: 128
6
+ seed: 0
7
+ dataset_shuffle_seed: 0
8
+ device: cuda
9
+ save_video: true
10
+ save_train_video: true
11
+ use_tb: true
12
+ use_wandb: true
13
+ wandb_run_id: '3469_0'
14
+ wandb_notes: 3469_0_req_3740_3restarted_3
15
+ eval: false
16
+ true_action_history: false
17
+ train_pad_after: 4
18
+ process_contact_features: ${eval}
19
+ obs_type: pixels
20
+ use_color: true
21
+ use_depth: true
22
+ use_masks: true
23
+ mask_list:
24
+ - EE_obj_mask
25
+ mask_representation: channels
26
+ crop_hw:
27
+ - 144
28
+ - 144
29
+ crop_down_offset: 48
30
+ color_crop_type: null
31
+ depth_crop_type: null
32
+ segmask_crop_type: null
33
+ add_crop_binary_mask: false
34
+ add_coord_conv_map: false
35
+ use_context_color: false
36
+ use_context_depth: false
37
+ use_context_segmask: false
38
+ context_color_crop_type: null
39
+ context_depth_crop_type: null
40
+ context_segmask_crop_type: null
41
+ context_add_crop_binary_mask: false
42
+ context_add_coord_conv_map: false
43
+ use_contact_map: false
44
+ use_sdf_maps: false
45
+ use_normals_maps: false
46
+ which_objects: both
47
+ max_contact_prob: 0.1
48
+ max_depth: 2.0
49
+ grasped_dtc_max_value: 0.2
50
+ env_dtc_max_value: 0.4
51
+ grasped_normals_mask_max_dtc_value: 0.2
52
+ env_normals_mask_max_dtc_value: 0.4
53
+ clamp_dtc: true
54
+ dtc_adaptive_normalization: false
55
+ mask_normals_within_sdf: true
56
+ adaptive_normals_mask: true
57
+ learnable_contact_preprocess_params: true
58
+ contact_model_name: local_multitask_outhd64all_home_crop_h144w144d48_mask_ctxtmask_seed_220979_epoch_9
59
+ contact_estimation_model_ckpt_path: ~/fish_leon/contact_estimation/artifacts/175604_2/checkpoints/epoch=09-val_loss=0.00.ckpt
60
+ encoder_type: small
61
+ debug_timestamps: false
62
+ open_loop: false
63
+ action_trajectories: true
64
+ stop_after_action: false
65
+ interpolation_frequency: 25
66
+ policy_frequency: 5
67
+ wait_for_new_camera_frames: true
68
+ baseline: false
69
+ train_demo_idxs_list_or_num: -1
70
+ log_train_every_steps: 25
71
+ name_of_expert_demo: 120_240x320_all_twodim_left_to_right_annotated_start_idx_5hz_zstd7_EE_pxl_coords_expert_demos_imp_act
72
+ expert_dataset_dirpath: ${root_dir}/FISH/expert_demos/${suite.name}/${task_name}/${name_of_expert_demo}
73
+ store_dataset_in_memory: false
74
+ expert_dataset: ${expert_dataset_dirpath}/demos.zarr
75
+ action_key: ${oc.if_else:${action_trajectories}, 'action_trajectory_${interpolation_frequency}hz',
76
+ 'action'}
77
+ semantic_demo_grouping_name: semantic_demo_grouping.yaml
78
+ semantic_demo_grouping: ${expert_dataset_dirpath}/${semantic_demo_grouping_name}
79
+ include_groups_list: all
80
+ expert_dataset_config: ${expert_dataset_dirpath}/demo_config.yaml
81
+ name_of_valid_demo: 120_240x320_all_twodim_left_to_right_annotated_start_idx_5hz_zstd7_EE_pxl_coords_expert_demos_imp_act
82
+ valid_dataset_dir: ${root_dir}/FISH/expert_demos/${suite.name}/${task_name}/${name_of_valid_demo}/demos.zarr
83
+ valid_demo_idxs_list_or_num: null
84
+ val_num_groups: 5
85
+ load_bc: ${agent.load_checkpoint}
86
+ checkpoint_epoch_list:
87
+ - 99
88
+ - 199
89
+ - 299
90
+ - 399
91
+ - 499
92
+ - 599
93
+ - 699
94
+ - 799
95
+ - 899
96
+ - 999
97
+ - 1249
98
+ - 1499
99
+ - 1749
100
+ - 1999
101
+ - 2999
102
+ - 3999
103
+ - 4999
104
+ - 5999
105
+ - 6999
106
+ - 7999
107
+ - 8999
108
+ - 9999
109
+ snapshot_root_dir: /mnt/grasp_high_usage/leonmkim/contact_estimation/FISH
110
+ save_snapshot: true
111
+ save_last_snapshot: true
112
+ save_snapshot_when_done: true
113
+ top_k_checkpoints: 5
114
+ save_snapshot_link_to_weights_dir: deprecated
115
+ restore_missing_checkpoint: null
116
+ bc_regularize: false
117
+ bc_weight_type: qfilter
118
+ experiment_dir: ./exp_local/${suite.name}_${obs_type}/${task_name}/${wandb_run_id}
119
+ agent:
120
+ _target_: agent.diffusion_policy.DiffusionPolicyAgent
121
+ name: diffusion_policy
122
+ load_checkpoint: ${eval}
123
+ device: ${device}
124
+ n_obs_steps: ${.config.policy_cfg.n_obs_steps}
125
+ suite_name: ${suite.name}
126
+ obs_type: ${obs_type}
127
+ enable_arm: ${eval}
128
+ enable_camera: ${eval}
129
+ use_tb: ${use_tb}
130
+ desired_image_shape:
131
+ - 13
132
+ - 180
133
+ - 240
134
+ orig_cam_shape:
135
+ - 3
136
+ - 240
137
+ - 320
138
+ config:
139
+ _target_: agent.diffusion_policy.DiffusionPolicyAgentConfig
140
+ compile: false
141
+ device: ${device}
142
+ cam_resize_shape: ${agent.desired_image_shape}
143
+ orig_cam_shape: ${agent.orig_cam_shape}
144
+ policy_frequency: ${policy_frequency}
145
+ interpolation_frequency: ${interpolation_frequency}
146
+ policy_cfg:
147
+ _target_: lerobot.common.policies.diffusion.configuration_diffusion.DiffusionConfig
148
+ n_obs_steps: 1
149
+ horizon: 36
150
+ n_action_steps: ${agent.config.policy_cfg.horizon}
151
+ output_shapes:
152
+ action:
153
+ - 7
154
+ input_normalization_modes:
155
+ observation.image: mean_std
156
+ observation.state: min_max
157
+ observation.action_history: min_max
158
+ output_normalization_modes:
159
+ action: min_max
160
+ vision_backbone: resnet18
161
+ crop_distractors_tlhw:
162
+ - 32
163
+ - 68
164
+ - 174
165
+ - 174
166
+ pretrained_backbone_weights: null
167
+ transforms:
168
+ - _target_: torchaug.transforms.RandomAffine
169
+ degrees:
170
+ - -5
171
+ - 5
172
+ translate:
173
+ - 0.05
174
+ - 0.05
175
+ batch_transform: true
176
+ num_chunks: -1
177
+ batch_inplace: true
178
+ - _target_: torchaug.transforms.RandomColorJitter
179
+ brightness: 0.3
180
+ contrast: 0.4
181
+ saturation: 0.5
182
+ hue: 0.08
183
+ batch_transform: true
184
+ num_chunks: -1
185
+ batch_inplace: true
186
+ use_group_norm: true
187
+ spatial_softmax_num_keypoints: 32
188
+ action_history_encoder_config:
189
+ _target_: lerobot.common.policies.diffusion.configuration_diffusion.Unet1dEncoderConfig
190
+ in_channels: 7
191
+ out_channels: 32
192
+ history_length: 6
193
+ kernel_size: ${agent.config.policy_cfg.kernel_size}
194
+ downsample_kernel_size: 3
195
+ downsample_stride: 2
196
+ downsample_padding: 1
197
+ down_dims:
198
+ - 256
199
+ - 512
200
+ - 1024
201
+ kernel_size: 5
202
+ n_groups: 8
203
+ diffusion_step_embed_dim: 128
204
+ use_film_scale_modulation: true
205
+ noise_scheduler_type: DDIM
206
+ beta_schedule: squaredcos_cap_v2
207
+ beta_start: 0.0001
208
+ beta_end: 0.02
209
+ prediction_type: epsilon
210
+ clip_sample: true
211
+ clip_sample_range: 1.0
212
+ num_train_timesteps: 50
213
+ num_inference_steps: 10
214
+ do_mask_loss_for_padding: false
215
+ input_shapes:
216
+ observation.image:
217
+ - 13
218
+ - 180
219
+ - 240
220
+ context_observation.image:
221
+ - 13
222
+ - 180
223
+ - 240
224
+ observation.state:
225
+ - 8
226
+ observation.action_history:
227
+ - 7
228
+ train_cfg:
229
+ _target_: utils.TrainConfig
230
+ lr: 0.0001
231
+ lr_scheduler: cosine
232
+ lr_warmup_steps: 500
233
+ adam_betas:
234
+ - 0.95
235
+ - 0.999
236
+ adam_eps: 1.0e-08
237
+ adam_weight_decay: 1.0e-06
238
+ grad_clip_norm: 10
239
+ offline_steps: ${num_train_frames_diffusion}
240
+ use_amp: true
241
+ observation_cfg:
242
+ _target_: agent.encoder.VisualFeatureSet
243
+ use_depth: ${use_depth}
244
+ use_color: ${use_color}
245
+ mask_input_dict:
246
+ _target_: agent.encoder.MaskInputDict
247
+ enable: ${use_masks}
248
+ representation: ${mask_representation}
249
+ mask_list: ${mask_list}
250
+ crop_input_config:
251
+ _target_: agent.encoder.CropInputConfig
252
+ color_crop_type: ${color_crop_type}
253
+ depth_crop_type: ${depth_crop_type}
254
+ segmask_crop_type: ${segmask_crop_type}
255
+ crop_hw: ${crop_hw}
256
+ crop_down_offset: ${crop_down_offset}
257
+ add_crop_binary_mask: ${add_crop_binary_mask}
258
+ add_coord_conv_map: ${add_coord_conv_map}
259
+ context_input_config:
260
+ _target_: agent.encoder.ContextInputConfig
261
+ use_color: ${use_context_color}
262
+ use_depth: ${use_context_depth}
263
+ mask_input_dict:
264
+ _target_: agent.encoder.MaskInputDict
265
+ enable: ${use_context_segmask}
266
+ representation: ${mask_representation}
267
+ mask_list: ${mask_list}
268
+ crop_input_config:
269
+ _target_: agent.encoder.CropInputConfig
270
+ color_crop_type: ${context_color_crop_type}
271
+ depth_crop_type: ${context_depth_crop_type}
272
+ segmask_crop_type: ${context_segmask_crop_type}
273
+ crop_hw: ${crop_hw}
274
+ crop_down_offset: ${crop_down_offset}
275
+ add_crop_binary_mask: ${context_add_crop_binary_mask}
276
+ add_coord_conv_map: ${context_add_coord_conv_map}
277
+ mask_soft_approx_scheduler_config:
278
+ _target_: agent.encoder.MaskSoftApproxSchedulerConfig
279
+ num_steps: 40000
280
+ initial_value: 10.0
281
+ final_value: 1000.0
282
+ interpolation_scheme: cosine
283
+ use_contact_map: ${use_contact_map}
284
+ use_sdf_maps: ${use_sdf_maps}
285
+ use_normals_maps: ${use_normals_maps}
286
+ which_objects: ${which_objects}
287
+ grasped_dtc_max_value: ${grasped_dtc_max_value}
288
+ env_dtc_max_value: ${env_dtc_max_value}
289
+ grasped_normals_mask_max_dtc_value: ${grasped_normals_mask_max_dtc_value}
290
+ env_normals_mask_max_dtc_value: ${env_normals_mask_max_dtc_value}
291
+ clamp_dtc: ${clamp_dtc}
292
+ max_contact_prob: ${max_contact_prob}
293
+ mask_normals_within_sdf: ${mask_normals_within_sdf}
294
+ dtc_adaptive_normalization: ${dtc_adaptive_normalization}
295
+ adaptive_normals_mask: ${adaptive_normals_mask}
296
+ max_depth: ${max_depth}
297
+ image_shape: ${agent.desired_image_shape}
298
+ learnable_contact_preprocess_params: ${learnable_contact_preprocess_params}
299
+ learning_rate: 0.0001
300
+ weight_decay: 0.0
301
+ contact_model_name: ${contact_model_name}
302
+ zero_centered: false
303
+ crop_distractors_tlhw:
304
+ - 32
305
+ - 68
306
+ - 174
307
+ - 174
308
+ suite:
309
+ suite: frankagym
310
+ name: frankagym
311
+ frame_stack: ${agent.n_obs_steps}
312
+ action_repeat: 1
313
+ discount: 0.99
314
+ hidden_dim: 1024
315
+ num_train_frames: 1000000
316
+ num_seed_frames: 0
317
+ num_train_epochs: 15000
318
+ validate_every_epochs: 250
319
+ validate_diffusion_on_action_loss_every_epochs: 250
320
+ train_eval_diffusion_on_action_loss_every_epochs: 250
321
+ check_topk_every_epochs: 250
322
+ save_snapshot_every_epochs: 1500
323
+ eval_every_frames: 2000
324
+ num_eval_episodes: 5
325
+ save_snapshot: true
326
+ wait_for_user_to_start_episode: true
327
+ task_make_fn:
328
+ _target_: suite.frankagym.make
329
+ name: ${task_name}
330
+ height: 240
331
+ width: 320
332
+ frame_stack: ${suite.frame_stack}
333
+ action_repeat: ${suite.action_repeat}
334
+ seed: ${seed}
335
+ enable_arm: ${agent.enable_arm}
336
+ enable_gripper: ${enable_gripper}
337
+ start_with_gripper_open: ${start_with_gripper_open}
338
+ enable_camera: ${agent.enable_camera}
339
+ path_to_depth_extrinsics: ${path_to_depth_extrinsics}
340
+ contact_estimation_model_ckpt_path: ${contact_estimation_model_ckpt_path}
341
+ x_limit: ${x_limit}
342
+ y_limit: ${y_limit}
343
+ z_limit: ${z_limit}
344
+ device: ${device}
345
+ interpolation_frequency: ${interpolation_frequency}
346
+ policy_frequency: ${policy_frequency}
347
+ debug_timestamps: ${debug_timestamps}
348
+ stop_after_action: ${stop_after_action}
349
+ open_loop: ${open_loop}
350
+ wait_for_new_camera_frames: ${wait_for_new_camera_frames}
351
+ action_key: ${action_key}
352
+ action_trajectory_horizon: ${agent.config.policy_cfg.horizon}
353
+ action_trajectories: ${action_trajectories}
354
+ path_to_zarr_dataset: ${expert_dataset}
355
+ agent_policy_cfg:
356
+ _target_: agent.diffusion_policy.DiffusionPolicyAgentConfig
357
+ compile: false
358
+ device: ${device}
359
+ cam_resize_shape: ${agent.desired_image_shape}
360
+ orig_cam_shape: ${agent.orig_cam_shape}
361
+ policy_frequency: ${policy_frequency}
362
+ interpolation_frequency: ${interpolation_frequency}
363
+ policy_cfg:
364
+ _target_: lerobot.common.policies.diffusion.configuration_diffusion.DiffusionConfig
365
+ n_obs_steps: 1
366
+ horizon: 36
367
+ n_action_steps: ${agent.config.policy_cfg.horizon}
368
+ output_shapes:
369
+ action:
370
+ - 7
371
+ input_normalization_modes:
372
+ observation.image: mean_std
373
+ observation.state: min_max
374
+ observation.action_history: min_max
375
+ output_normalization_modes:
376
+ action: min_max
377
+ vision_backbone: resnet18
378
+ crop_distractors_tlhw:
379
+ - 32
380
+ - 68
381
+ - 174
382
+ - 174
383
+ pretrained_backbone_weights: null
384
+ transforms:
385
+ - _target_: torchaug.transforms.RandomAffine
386
+ degrees:
387
+ - -5
388
+ - 5
389
+ translate:
390
+ - 0.05
391
+ - 0.05
392
+ batch_transform: true
393
+ num_chunks: -1
394
+ batch_inplace: true
395
+ - _target_: torchaug.transforms.RandomColorJitter
396
+ brightness: 0.3
397
+ contrast: 0.4
398
+ saturation: 0.5
399
+ hue: 0.08
400
+ batch_transform: true
401
+ num_chunks: -1
402
+ batch_inplace: true
403
+ use_group_norm: true
404
+ spatial_softmax_num_keypoints: 32
405
+ action_history_encoder_config:
406
+ _target_: lerobot.common.policies.diffusion.configuration_diffusion.Unet1dEncoderConfig
407
+ in_channels: 7
408
+ out_channels: 32
409
+ history_length: 6
410
+ kernel_size: ${agent.config.policy_cfg.kernel_size}
411
+ downsample_kernel_size: 3
412
+ downsample_stride: 2
413
+ downsample_padding: 1
414
+ down_dims:
415
+ - 256
416
+ - 512
417
+ - 1024
418
+ kernel_size: 5
419
+ n_groups: 8
420
+ diffusion_step_embed_dim: 128
421
+ use_film_scale_modulation: true
422
+ noise_scheduler_type: DDIM
423
+ beta_schedule: squaredcos_cap_v2
424
+ beta_start: 0.0001
425
+ beta_end: 0.02
426
+ prediction_type: epsilon
427
+ clip_sample: true
428
+ clip_sample_range: 1.0
429
+ num_train_timesteps: 50
430
+ num_inference_steps: 10
431
+ do_mask_loss_for_padding: false
432
+ input_shapes:
433
+ observation.image:
434
+ - 13
435
+ - 180
436
+ - 240
437
+ context_observation.image:
438
+ - 13
439
+ - 180
440
+ - 240
441
+ observation.state:
442
+ - 8
443
+ observation.action_history:
444
+ - 7
445
+ train_cfg:
446
+ _target_: utils.TrainConfig
447
+ lr: 0.0001
448
+ lr_scheduler: cosine
449
+ lr_warmup_steps: 500
450
+ adam_betas:
451
+ - 0.95
452
+ - 0.999
453
+ adam_eps: 1.0e-08
454
+ adam_weight_decay: 1.0e-06
455
+ grad_clip_norm: 10
456
+ offline_steps: ${num_train_frames_diffusion}
457
+ use_amp: true
458
+ observation_cfg:
459
+ _target_: agent.encoder.VisualFeatureSet
460
+ use_depth: ${use_depth}
461
+ use_color: ${use_color}
462
+ mask_input_dict:
463
+ _target_: agent.encoder.MaskInputDict
464
+ enable: ${use_masks}
465
+ representation: ${mask_representation}
466
+ mask_list: ${mask_list}
467
+ crop_input_config:
468
+ _target_: agent.encoder.CropInputConfig
469
+ color_crop_type: ${color_crop_type}
470
+ depth_crop_type: ${depth_crop_type}
471
+ segmask_crop_type: ${segmask_crop_type}
472
+ crop_hw: ${crop_hw}
473
+ crop_down_offset: ${crop_down_offset}
474
+ add_crop_binary_mask: ${add_crop_binary_mask}
475
+ add_coord_conv_map: ${add_coord_conv_map}
476
+ context_input_config:
477
+ _target_: agent.encoder.ContextInputConfig
478
+ use_color: ${use_context_color}
479
+ use_depth: ${use_context_depth}
480
+ mask_input_dict:
481
+ _target_: agent.encoder.MaskInputDict
482
+ enable: ${use_context_segmask}
483
+ representation: ${mask_representation}
484
+ mask_list: ${mask_list}
485
+ crop_input_config:
486
+ _target_: agent.encoder.CropInputConfig
487
+ color_crop_type: ${context_color_crop_type}
488
+ depth_crop_type: ${context_depth_crop_type}
489
+ segmask_crop_type: ${context_segmask_crop_type}
490
+ crop_hw: ${crop_hw}
491
+ crop_down_offset: ${crop_down_offset}
492
+ add_crop_binary_mask: ${context_add_crop_binary_mask}
493
+ add_coord_conv_map: ${context_add_coord_conv_map}
494
+ mask_soft_approx_scheduler_config:
495
+ _target_: agent.encoder.MaskSoftApproxSchedulerConfig
496
+ num_steps: 40000
497
+ initial_value: 10.0
498
+ final_value: 1000.0
499
+ interpolation_scheme: cosine
500
+ use_contact_map: ${use_contact_map}
501
+ use_sdf_maps: ${use_sdf_maps}
502
+ use_normals_maps: ${use_normals_maps}
503
+ which_objects: ${which_objects}
504
+ grasped_dtc_max_value: ${grasped_dtc_max_value}
505
+ env_dtc_max_value: ${env_dtc_max_value}
506
+ grasped_normals_mask_max_dtc_value: ${grasped_normals_mask_max_dtc_value}
507
+ env_normals_mask_max_dtc_value: ${env_normals_mask_max_dtc_value}
508
+ clamp_dtc: ${clamp_dtc}
509
+ max_contact_prob: ${max_contact_prob}
510
+ mask_normals_within_sdf: ${mask_normals_within_sdf}
511
+ dtc_adaptive_normalization: ${dtc_adaptive_normalization}
512
+ adaptive_normals_mask: ${adaptive_normals_mask}
513
+ max_depth: ${max_depth}
514
+ image_shape: ${agent.desired_image_shape}
515
+ learnable_contact_preprocess_params: ${learnable_contact_preprocess_params}
516
+ learning_rate: 0.0001
517
+ weight_decay: 0.0
518
+ contact_model_name: ${contact_model_name}
519
+ zero_centered: false
520
+ crop_distractors_tlhw:
521
+ - 32
522
+ - 68
523
+ - 174
524
+ - 174
525
+ true_action_history: ${true_action_history}
526
+ num_train_frames_bc: 50000
527
+ num_train_frames_drq: 1100000
528
+ stddev_schedule_drq: linear(1.0,0.1,100000)
529
+ task_name: FrankaInsertion-v1
530
+ num_train_frames_vinn: 25000
531
+ num_train_frames_diffusion: 1000000
532
+ num_train_epochs_bc: 5000
533
+ num_train_epochs_diffusion: 15000
534
+ validate_every_epochs_bc: 5
535
+ validate_every_epochs_diffusion: 250
536
+ validate_diffusion_on_action_loss_every_epochs: 250
537
+ train_eval_diffusion_on_action_loss_every_epochs: 250
538
+ check_topk_every_epochs: 5
539
+ check_topk_every_epochs_diffusion: 250
540
+ save_snapshot_every_epochs_diffusion: 1500
541
+ x_limit:
542
+ - 0.2
543
+ - 0.7
544
+ y_limit:
545
+ - -0.4
546
+ - 0.4
547
+ z_limit:
548
+ - -0.05
549
+ - 0.55
550
+ home_displacement:
551
+ - 0.55
552
+ - 0.0
553
+ - 0.55
554
+ - 180.0
555
+ - 0.0
556
+ - 0.0
557
+ enable_gripper: true
558
+ start_with_gripper_open: true
559
+ offset_mask:
560
+ - 1
561
+ - 1
562
+ - 1
563
+ - 1
564
+ - 1
565
+ - 1
566
+ path_to_depth_extrinsics: ~/fish_leon/FISH/cfgs/camera_poses/camera_poses_L515/20240904-122305/color_tf_world.npy
567
+ test:
568
+ username: leonmkim
569
+ hostname: grasp-login1
570
+ name_of_expert_demo: 112_240x320_all_twodim_left_to_right_annotated_start_idx_5hz_zstd7_EE_pxl_coords_expert_demos_imp_act
571
+ root_dir: ~/fish_leon
572
+ expert_dataset_dirpath: ${test.root_dir}/FISH/expert_demos/${suite.name}/${task_name}/${test.name_of_expert_demo}
573
+ expert_dataset: ${test.expert_dataset_dirpath}/demos.zarr
574
+ semantic_demo_grouping_name: semantic_demo_grouping.yaml
575
+ semantic_demo_grouping: ${test.expert_dataset_dirpath}/${test.semantic_demo_grouping_name}
576
+ include_groups_list: all
577
+ batch_size: 128
578
+ store_dataset_in_memory: false
579
+ use_tb: true
580
+ local_snapshot_root_dir: /mnt/bighdd/fish_contact_backup
581
+ local_snapshot_dir: ${test.local_snapshot_root_dir}/exp_local/${suite.name}_${obs_type}/${task_name}
582
+ resume_wandb_run: false
583
+ feature_type: 180x240_crpdstlhw24x51x130x130_1_RGB_D_2.0_msk_channels_EE_obj_mask_acthist_hst6_out32_dwnkrnl3_dwnstrd2_dwnpd1
snapshot_10500.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:717b0e40102bb205fe502cfc01299c1b28995e10d22c1ebe630b4b27ca536473
3
+ size 910102678
snapshot_10999.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b1b2fcd6eb1204fdd7ffa80d0316516a982781186e3caecda698b049dc6ada58
3
+ size 910102678
snapshot_11499.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9c1f8463e39e94045c3da74f06306ac92f2fd872c35f54907275b1ef98d23bb1
3
+ size 910102678
snapshot_11749.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2baa85aec989cbeb5840dd7822443f0319589de9de70323cc328efc5f9107045
3
+ size 910102678
snapshot_11999.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:50fde690ad34dc7f722268eab0ed92295d314592673fb3151cda1ff8da2bb0f8
3
+ size 910102678
snapshot_12000.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5c217bf6289e90fc3708050342ddfca476d0bf41e04171eeb1e6cfee9a8773d8
3
+ size 910102678
snapshot_13500.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7a5a55af4d58beae6291e81025ecf2b5e5c5428048c77bc3d5a6ed29ce046ea5
3
+ size 910102678
snapshot_14749.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:07bc3a11a9ae0167e19903e622163b95a3d7381385b6bbd42502111c28b58d50
3
+ size 910102678
snapshot_15000.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9f3b51c0f5be86e65abe46caf62c663084c1f3d91eb64afe51a6d0aa7c490997
3
+ size 910102678
snapshot_4500.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ff55ce8a48d5aa06a68c4aa85c4bfa9b2d56264143b5559c83158dba4eb708e3
3
+ size 910102678
snapshot_5499.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:56305b4bdbc6e419ff7d91bb2bc50e05182c0b3ddb4ea2cec3b3a3457100fc2a
3
+ size 910102678
snapshot_5999.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ca39a835d1a5605e437481ad03dcd97f72f593b9e3aa860feed1e0de96b6ebe8
3
+ size 910102678
snapshot_6000.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e36e3319cacf466694912e4dcc365fca1c495ac5ce94b3af27f66a7123284131
3
+ size 910102678
snapshot_7500.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2e77b40dbda4e17141abcc1e7a2e4947d55242c5bd4ec152ee44ae37dcdbbefa
3
+ size 910102678
snapshot_9000.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:10d3b745fbdae41d96424ef68644d38478305576c22eb67d1f90e0baa1f153ae
3
+ size 910102678
snapshot_9749.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ecfeb069e37bd45cda91c47293339577417efe2dc66cc1830bcd7fd889b8fb33
3
+ size 910102678
snapshot_last.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9f3b51c0f5be86e65abe46caf62c663084c1f3d91eb64afe51a6d0aa7c490997
3
+ size 910102678