serialexperimentsleon commited on
Commit
96c0f92
·
verified ·
1 Parent(s): ae2568f

Add files using upload-large-folder tool

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