serialexperimentsleon commited on
Commit
bc12d2a
·
verified ·
1 Parent(s): 03abb13

Add files using upload-large-folder tool

Browse files
This view is limited to 50 files because it contains too many changes.   See raw diff
Files changed (50) hide show
  1. .gitattributes +20 -0
  2. 204954/.hydra/config.yaml +349 -0
  3. 204954/.hydra/hydra.yaml +169 -0
  4. 204954/.hydra/overrides.yaml +3 -0
  5. 204954/eval_policy.log +15 -0
  6. 205123/.hydra/config.yaml +349 -0
  7. 205123/.hydra/hydra.yaml +169 -0
  8. 205123/.hydra/overrides.yaml +3 -0
  9. 205123/episode_rosbags/aligned_depth_to_color_K.npy +3 -0
  10. 205123/episode_rosbags/cam_tf_world.npy +3 -0
  11. 205123/episode_rosbags/color_K.npy +3 -0
  12. 205123/episode_rosbags/depth_K.npy +3 -0
  13. 205123/eval_robot.log +12 -0
  14. 205123/eval_video/0_eval.mp4 +3 -0
  15. 205123/tb/events.out.tfevents.1733881891.leonmkim-ROG-Strix-G15CS-G15CS.596892.0 +3 -0
  16. 205123/wandb/debug-internal.log +351 -0
  17. 205123/wandb/debug.log +31 -0
  18. 205123/wandb/run-20241210_205130-svysfc56/files/code/FISH/eval_robot.py +512 -0
  19. 205123/wandb/run-20241210_205130-svysfc56/files/config.yaml +966 -0
  20. 205123/wandb/run-20241210_205130-svysfc56/files/diff.patch +25 -0
  21. 205123/wandb/run-20241210_205130-svysfc56/files/media/videos/eval/video_0_b6b0b3ba17b7513a5df9.mp4 +3 -0
  22. 205123/wandb/run-20241210_205130-svysfc56/files/output.log +1049 -0
  23. 205123/wandb/run-20241210_205130-svysfc56/files/requirements.txt +339 -0
  24. 205123/wandb/run-20241210_205130-svysfc56/files/wandb-metadata.json +92 -0
  25. 205123/wandb/run-20241210_205130-svysfc56/files/wandb-summary.json +1 -0
  26. 205123/wandb/run-20241210_205130-svysfc56/logs/debug-internal.log +351 -0
  27. 205123/wandb/run-20241210_205130-svysfc56/logs/debug.log +31 -0
  28. 205123/wandb/run-20241210_205130-svysfc56/run-svysfc56.wandb +0 -0
  29. 205307/.hydra/config.yaml +349 -0
  30. 205307/.hydra/hydra.yaml +169 -0
  31. 205307/eval_policy.log +15 -0
  32. 205324/.hydra/config.yaml +349 -0
  33. 205324/.hydra/hydra.yaml +169 -0
  34. 205324/.hydra/overrides.yaml +3 -0
  35. 205324/episode_rosbags/aligned_depth_to_color_K.npy +3 -0
  36. 205324/episode_rosbags/cam_tf_world.npy +3 -0
  37. 205324/episode_rosbags/color_K.npy +3 -0
  38. 205324/episode_rosbags/depth_K.npy +3 -0
  39. 205324/eval_robot.log +12 -0
  40. 205324/wandb/debug-internal.log +327 -0
  41. 205324/wandb/debug.log +31 -0
  42. 205324/wandb/run-20241210_205330-52nddla7/files/code/FISH/eval_robot.py +512 -0
  43. 205324/wandb/run-20241210_205330-52nddla7/files/config.yaml +966 -0
  44. 205324/wandb/run-20241210_205330-52nddla7/files/diff.patch +34 -0
  45. 205324/wandb/run-20241210_205330-52nddla7/files/output.log +461 -0
  46. 205324/wandb/run-20241210_205330-52nddla7/files/requirements.txt +339 -0
  47. 205324/wandb/run-20241210_205330-52nddla7/files/wandb-metadata.json +92 -0
  48. 205324/wandb/run-20241210_205330-52nddla7/files/wandb-summary.json +1 -0
  49. 205324/wandb/run-20241210_205330-52nddla7/logs/debug-internal.log +327 -0
  50. 205324/wandb/run-20241210_205330-52nddla7/logs/debug.log +31 -0
.gitattributes CHANGED
@@ -33,3 +33,23 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
36
+ 205123/wandb/run-20241210_205130-svysfc56/files/media/videos/eval/video_0_b6b0b3ba17b7513a5df9.mp4 filter=lfs diff=lfs merge=lfs -text
37
+ 205123/eval_video/0_eval.mp4 filter=lfs diff=lfs merge=lfs -text
38
+ 205451/eval_video/0_eval.mp4 filter=lfs diff=lfs merge=lfs -text
39
+ 205451/wandb/run-20241210_205457-52qyo5rz/files/media/videos/eval/video_0_bfb699cd8c28a00c2c34.mp4 filter=lfs diff=lfs merge=lfs -text
40
+ 205451/episode_rosbags/episode_0_2024-12-10-20-55-17.bag filter=lfs diff=lfs merge=lfs -text
41
+ 214032/episode_rosbags/episode_2_2024-12-10-21-42-51.rrd filter=lfs diff=lfs merge=lfs -text
42
+ 214032/episode_rosbags/episode_0_2024-12-10-21-41-00.rrd filter=lfs diff=lfs merge=lfs -text
43
+ 214032/episode_rosbags/episode_1_2024-12-10-21-41-44.rrd filter=lfs diff=lfs merge=lfs -text
44
+ 214032/episode_rosbags/episode_4_2024-12-10-21-44-14.rrd filter=lfs diff=lfs merge=lfs -text
45
+ 214032/wandb/run-20241210_214038-s9hpsjf9/run-s9hpsjf9.wandb filter=lfs diff=lfs merge=lfs -text
46
+ 214032/wandb/run-20241210_214038-s9hpsjf9/files/media/videos/eval/4_eval_4_86ec182da7a62ff7b578.mp4 filter=lfs diff=lfs merge=lfs -text
47
+ 214032/wandb/run-20241210_214038-s9hpsjf9/files/media/videos/eval/0_eval_0_99cde89795fdf8f2318a.mp4 filter=lfs diff=lfs merge=lfs -text
48
+ 214032/wandb/run-20241210_214038-s9hpsjf9/files/media/videos/eval/2_eval_2_aac838f24394620216df.mp4 filter=lfs diff=lfs merge=lfs -text
49
+ 214032/eval_video/0_eval.mp4 filter=lfs diff=lfs merge=lfs -text
50
+ 214032/wandb/run-20241210_214038-s9hpsjf9/files/media/videos/eval/1_eval_1_0f1a83ed932af73a96be.mp4 filter=lfs diff=lfs merge=lfs -text
51
+ 214032/wandb/run-20241210_214038-s9hpsjf9/files/media/videos/eval/3_eval_3_9f3ddf93c395db4f718a.mp4 filter=lfs diff=lfs merge=lfs -text
52
+ 214032/eval_video/3_eval.mp4 filter=lfs diff=lfs merge=lfs -text
53
+ 214032/eval_video/1_eval.mp4 filter=lfs diff=lfs merge=lfs -text
54
+ 214032/eval_video/2_eval.mp4 filter=lfs diff=lfs merge=lfs -text
55
+ 214032/eval_video/4_eval.mp4 filter=lfs diff=lfs merge=lfs -text
204954/.hydra/config.yaml ADDED
@@ -0,0 +1,349 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ root_dir: /home/${oc.env:USER}/fish_leon
2
+ nstep: 3
3
+ seed: 41
4
+ dataset_shuffle_seed: ${seed}
5
+ device: cuda
6
+ save_video: true
7
+ save_buffer: true
8
+ use_tb: true
9
+ baseline: false
10
+ use_wandb: true
11
+ eval: true
12
+ process_contact_features: ${eval}
13
+ obs_type: pixels
14
+ use_color: true
15
+ use_depth: true
16
+ use_masks: false
17
+ mask_list:
18
+ - EE_obj_mask
19
+ mask_representation: channels
20
+ crop_hw:
21
+ - 144
22
+ - 144
23
+ crop_down_offset: 48
24
+ color_crop_type: null
25
+ depth_crop_type: null
26
+ segmask_crop_type: null
27
+ add_crop_binary_mask: false
28
+ add_coord_conv_map: false
29
+ use_context_color: false
30
+ use_context_depth: false
31
+ use_context_segmask: false
32
+ context_color_crop_type: null
33
+ context_depth_crop_type: null
34
+ context_segmask_crop_type: null
35
+ context_add_crop_binary_mask: false
36
+ context_add_coord_conv_map: false
37
+ use_contact_map: false
38
+ use_sdf_maps: false
39
+ use_normals_maps: false
40
+ which_objects: both
41
+ max_contact_prob: 0.1
42
+ max_depth: 2.0
43
+ grasped_dtc_max_value: 0.105
44
+ env_dtc_max_value: 0.425
45
+ grasped_normals_mask_max_dtc_value: 0.105
46
+ env_normals_mask_max_dtc_value: 0.425
47
+ clamp_dtc: true
48
+ dtc_adaptive_normalization: false
49
+ mask_normals_within_sdf: true
50
+ adaptive_normals_mask: true
51
+ learnable_contact_preprocess_params: false
52
+ contact_model_name: local_multitask_outhd64all_home_crop_h144w144d48_ctxt_seed_183386_epoch_9
53
+ contact_estimation_model_ckpt_path: ~/fish_leon/contact_estimation/artifacts/175604_2/checkpoints/epoch=09-val_loss=0.00.ckpt
54
+ num_eval: 5
55
+ debug_timestamps: false
56
+ open_loop: false
57
+ action_trajectories: true
58
+ stop_after_action: false
59
+ interpolation_frequency: 25
60
+ policy_frequency: 5
61
+ wait_for_new_camera_frames: true
62
+ random_start: false
63
+ eval_starts: ${root_dir}/FISH/eval_starts/${suite.name}_${obs_type}/${task_name}
64
+ train_demo_idxs_list_or_num: null
65
+ num_valid_demos: null
66
+ val_num_groups: 3
67
+ name_of_expert_demo: 192_240x320_multiobj_twodim_recovery_5hz_zstd7_EE_pxl_coords_expert_demos_imp_act
68
+ expert_dataset_dirpath: ${root_dir}/FISH/expert_demos/${suite.name}/${task_name}/${name_of_expert_demo}
69
+ expert_dataset: ${expert_dataset_dirpath}/demos.zarr
70
+ action_key: ${oc.if_else:${action_trajectories}, 'action_trajectory_${interpolation_frequency}hz',
71
+ 'action'}
72
+ semantic_demo_grouping_name: semantic_demo_grouping.yaml
73
+ semantic_demo_grouping: ${expert_dataset_dirpath}/${semantic_demo_grouping_name}
74
+ expert_dataset_config: ${expert_dataset_dirpath}/demo_config.yaml
75
+ bc_regularize: false
76
+ bc_weight_type: qfilter
77
+ load_checkpoint: ${agent.load_checkpoint}
78
+ wandb_run_id: '265626'
79
+ true_action_history: true
80
+ wandb_notes: null
81
+ checkpoint_epoch: 12000
82
+ load_residual_weight: false
83
+ checkpoint_root_dir: /home/${oc.env:USER}/fish_leon/FISH
84
+ checkpoint_weight_dir: ${checkpoint_root_dir}/exp_local/${suite.name}_${obs_type}/${task_name}/${wandb_run_id}
85
+ residual_weight: ${root_dir}/FISH/weights/${suite.name}_${obs_type}/${task_name}/weight.pt
86
+ experiment_dir: ./exp_local/${suite.name}_${obs_type}/${task_name}/${wandb_run_id}
87
+ agent:
88
+ _target_: agent.diffusion_policy.DiffusionPolicyAgent
89
+ name: diffusion_policy
90
+ load_checkpoint: ${eval}
91
+ device: ${device}
92
+ n_obs_steps: ${.config.policy_cfg.n_obs_steps}
93
+ suite_name: ${suite.name}
94
+ obs_type: ${obs_type}
95
+ enable_arm: ${eval}
96
+ enable_camera: ${eval}
97
+ use_tb: ${use_tb}
98
+ desired_image_shape:
99
+ - 13
100
+ - 180
101
+ - 240
102
+ orig_cam_shape:
103
+ - 3
104
+ - 240
105
+ - 320
106
+ config:
107
+ _target_: agent.diffusion_policy.DiffusionPolicyAgentConfig
108
+ compile: false
109
+ device: ${device}
110
+ cam_resize_shape: ${agent.desired_image_shape}
111
+ orig_cam_shape: ${agent.orig_cam_shape}
112
+ policy_frequency: ${policy_frequency}
113
+ interpolation_frequency: ${interpolation_frequency}
114
+ policy_cfg:
115
+ _target_: lerobot.common.policies.diffusion.configuration_diffusion.DiffusionConfig
116
+ n_obs_steps: 1
117
+ horizon: 36
118
+ n_action_steps: ${agent.config.policy_cfg.horizon}
119
+ input_shapes:
120
+ observation.image: ${agent.config.cam_resize_shape}
121
+ context_observation.image: ${agent.config.cam_resize_shape}
122
+ observation.state:
123
+ - 8
124
+ observation.action_history:
125
+ - 7
126
+ output_shapes:
127
+ action:
128
+ - 7
129
+ input_normalization_modes:
130
+ observation.image: mean_std
131
+ observation.state: min_max
132
+ observation.action_history: min_max
133
+ output_normalization_modes:
134
+ action: min_max
135
+ vision_backbone: resnet18
136
+ pretrained_backbone_weights: null
137
+ transforms:
138
+ - _target_: torchaug.transforms.RandomAffine
139
+ degrees:
140
+ - -5
141
+ - 5
142
+ translate:
143
+ - 0.05
144
+ - 0.05
145
+ batch_transform: true
146
+ num_chunks: -1
147
+ batch_inplace: true
148
+ - _target_: torchaug.transforms.RandomColorJitter
149
+ brightness: 0.3
150
+ contrast: 0.4
151
+ saturation: 0.5
152
+ hue: 0.08
153
+ batch_transform: true
154
+ num_chunks: -1
155
+ batch_inplace: true
156
+ use_group_norm: true
157
+ spatial_softmax_num_keypoints: 32
158
+ action_history_encoder_config:
159
+ _target_: lerobot.common.policies.diffusion.configuration_diffusion.Unet1dEncoderConfig
160
+ in_channels: 7
161
+ out_channels: 32
162
+ history_length: ${agent.config.policy_cfg.n_action_steps}
163
+ kernel_size: ${agent.config.policy_cfg.kernel_size}
164
+ downsample_kernel_size: 3
165
+ downsample_stride: 2
166
+ downsample_padding: 1
167
+ down_dims:
168
+ - 256
169
+ - 512
170
+ - 1024
171
+ kernel_size: 5
172
+ n_groups: 8
173
+ diffusion_step_embed_dim: 128
174
+ use_film_scale_modulation: true
175
+ noise_scheduler_type: DDIM
176
+ beta_schedule: squaredcos_cap_v2
177
+ beta_start: 0.0001
178
+ beta_end: 0.02
179
+ prediction_type: epsilon
180
+ clip_sample: true
181
+ clip_sample_range: 1.0
182
+ num_train_timesteps: 50
183
+ num_inference_steps: 10
184
+ do_mask_loss_for_padding: false
185
+ train_cfg:
186
+ _target_: utils.TrainConfig
187
+ lr: 0.0001
188
+ lr_scheduler: cosine
189
+ lr_warmup_steps: 500
190
+ adam_betas:
191
+ - 0.95
192
+ - 0.999
193
+ adam_eps: 1.0e-08
194
+ adam_weight_decay: 1.0e-06
195
+ grad_clip_norm: 10
196
+ offline_steps: ${num_train_frames_diffusion}
197
+ use_amp: true
198
+ observation_cfg:
199
+ _target_: agent.encoder.VisualFeatureSet
200
+ use_depth: ${use_depth}
201
+ use_color: ${use_color}
202
+ mask_input_dict:
203
+ _target_: agent.encoder.MaskInputDict
204
+ enable: ${use_masks}
205
+ representation: ${mask_representation}
206
+ mask_list: ${mask_list}
207
+ crop_input_config:
208
+ _target_: agent.encoder.CropInputConfig
209
+ color_crop_type: ${color_crop_type}
210
+ depth_crop_type: ${depth_crop_type}
211
+ segmask_crop_type: ${segmask_crop_type}
212
+ crop_hw: ${crop_hw}
213
+ crop_down_offset: ${crop_down_offset}
214
+ add_crop_binary_mask: ${add_crop_binary_mask}
215
+ add_coord_conv_map: ${add_coord_conv_map}
216
+ context_input_config:
217
+ _target_: agent.encoder.ContextInputConfig
218
+ use_color: ${use_context_color}
219
+ use_depth: ${use_context_depth}
220
+ mask_input_dict:
221
+ _target_: agent.encoder.MaskInputDict
222
+ enable: ${use_context_segmask}
223
+ representation: ${mask_representation}
224
+ mask_list: ${mask_list}
225
+ crop_input_config:
226
+ _target_: agent.encoder.CropInputConfig
227
+ color_crop_type: ${context_color_crop_type}
228
+ depth_crop_type: ${context_depth_crop_type}
229
+ segmask_crop_type: ${context_segmask_crop_type}
230
+ crop_hw: ${crop_hw}
231
+ crop_down_offset: ${crop_down_offset}
232
+ add_crop_binary_mask: ${context_add_crop_binary_mask}
233
+ add_coord_conv_map: ${context_add_coord_conv_map}
234
+ mask_soft_approx_scheduler_config:
235
+ _target_: agent.encoder.MaskSoftApproxSchedulerConfig
236
+ num_steps: 40000
237
+ initial_value: 10.0
238
+ final_value: 1000.0
239
+ interpolation_scheme: constant
240
+ use_contact_map: ${use_contact_map}
241
+ use_sdf_maps: ${use_sdf_maps}
242
+ use_normals_maps: ${use_normals_maps}
243
+ which_objects: ${which_objects}
244
+ grasped_dtc_max_value: ${grasped_dtc_max_value}
245
+ env_dtc_max_value: ${env_dtc_max_value}
246
+ grasped_normals_mask_max_dtc_value: ${grasped_normals_mask_max_dtc_value}
247
+ env_normals_mask_max_dtc_value: ${env_normals_mask_max_dtc_value}
248
+ clamp_dtc: ${clamp_dtc}
249
+ max_contact_prob: ${max_contact_prob}
250
+ mask_normals_within_sdf: ${mask_normals_within_sdf}
251
+ dtc_adaptive_normalization: ${dtc_adaptive_normalization}
252
+ adaptive_normals_mask: ${adaptive_normals_mask}
253
+ max_depth: ${max_depth}
254
+ image_shape: ${agent.desired_image_shape}
255
+ learnable_contact_preprocess_params: ${learnable_contact_preprocess_params}
256
+ learning_rate: ${agent.config.train_cfg.lr}
257
+ weight_decay: 0.0
258
+ contact_model_name: ${contact_model_name}
259
+ zero_centered: false
260
+ suite:
261
+ suite: frankagym
262
+ name: frankagym
263
+ frame_stack: ${agent.n_obs_steps}
264
+ action_repeat: 1
265
+ discount: 0.99
266
+ hidden_dim: 1024
267
+ num_train_frames: 2010
268
+ num_seed_frames: 260
269
+ num_train_epochs: 5000
270
+ validate_every_epochs: 100
271
+ validate_diffusion_on_action_loss_every_epochs: 500
272
+ train_eval_diffusion_on_action_loss_every_epochs: 500
273
+ check_topk_every_epochs: 10
274
+ save_snapshot_every_epochs: 5000
275
+ eval_every_frames: 2000
276
+ num_eval_episodes: 5
277
+ save_snapshot: true
278
+ wait_for_user_to_start_episode: true
279
+ task_make_fn:
280
+ _target_: suite.frankagym.make
281
+ name: ${task_name}
282
+ height: 240
283
+ width: 320
284
+ frame_stack: ${suite.frame_stack}
285
+ action_repeat: ${suite.action_repeat}
286
+ seed: ${seed}
287
+ enable_arm: ${agent.enable_arm}
288
+ enable_gripper: ${enable_gripper}
289
+ start_with_gripper_open: ${start_with_gripper_open}
290
+ enable_camera: ${agent.enable_camera}
291
+ path_to_depth_extrinsics: ${path_to_depth_extrinsics}
292
+ contact_estimation_model_ckpt_path: ${contact_estimation_model_ckpt_path}
293
+ x_limit: ${x_limit}
294
+ y_limit: ${y_limit}
295
+ z_limit: ${z_limit}
296
+ device: ${device}
297
+ interpolation_frequency: ${interpolation_frequency}
298
+ policy_frequency: ${policy_frequency}
299
+ debug_timestamps: ${debug_timestamps}
300
+ stop_after_action: ${stop_after_action}
301
+ open_loop: ${open_loop}
302
+ wait_for_new_camera_frames: ${wait_for_new_camera_frames}
303
+ action_key: ${action_key}
304
+ action_trajectory_horizon: ${agent.config.policy_cfg.horizon}
305
+ action_trajectories: ${action_trajectories}
306
+ path_to_zarr_dataset: ${expert_dataset}
307
+ agent_policy_cfg: ???
308
+ true_action_history: ${true_action_history}
309
+ num_train_frames_bc: 50000
310
+ num_train_frames_drq: 1100000
311
+ stddev_schedule_drq: linear(1.0,0.1,100000)
312
+ task_name: FrankaInsertion-v1
313
+ num_train_frames_vinn: 25000
314
+ num_train_frames_diffusion: 1000000
315
+ num_train_epochs_bc: 5000
316
+ num_train_epochs_diffusion: 5000
317
+ validate_every_epochs_bc: 5
318
+ validate_every_epochs_diffusion: 25
319
+ validate_diffusion_on_action_loss_every_epochs: 50
320
+ train_eval_diffusion_on_action_loss_every_epochs: 500
321
+ check_topk_every_epochs: 5
322
+ check_topk_every_epochs_diffusion: ${validate_diffusion_on_action_loss_every_epochs}
323
+ save_snapshot_every_epochs_diffusion: 5000
324
+ x_limit:
325
+ - 0.2
326
+ - 0.7
327
+ y_limit:
328
+ - -0.4
329
+ - 0.4
330
+ z_limit:
331
+ - -0.05
332
+ - 0.55
333
+ home_displacement:
334
+ - 0.55
335
+ - 0.0
336
+ - 0.55
337
+ - 180.0
338
+ - 0.0
339
+ - 0.0
340
+ enable_gripper: true
341
+ start_with_gripper_open: true
342
+ offset_mask:
343
+ - 1
344
+ - 1
345
+ - 1
346
+ - 1
347
+ - 1
348
+ - 1
349
+ path_to_depth_extrinsics: ~/fish_leon/FISH/cfgs/camera_poses/camera_poses_L515/20240904-122305/color_tf_world.npy
204954/.hydra/hydra.yaml ADDED
@@ -0,0 +1,169 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ hydra:
2
+ run:
3
+ dir: ${experiment_dir}/${now:%H%M%S}
4
+ sweep:
5
+ dir: ${experiment_dir}/${now:%H%M%S}
6
+ subdir: ${hydra.job.num}
7
+ launcher:
8
+ submitit_folder: ${experiment_dir}/${now:%H%M%S}/.slurm
9
+ timeout_min: 60
10
+ cpus_per_task: null
11
+ gpus_per_node: null
12
+ tasks_per_node: 1
13
+ mem_gb: null
14
+ nodes: 1
15
+ name: ${hydra.job.name}
16
+ stderr_to_stdout: false
17
+ _target_: hydra_plugins.hydra_submitit_launcher.submitit_launcher.LocalLauncher
18
+ sweeper:
19
+ _target_: hydra._internal.core_plugins.basic_sweeper.BasicSweeper
20
+ max_batch_size: null
21
+ params: null
22
+ help:
23
+ app_name: ${hydra.job.name}
24
+ header: '${hydra.help.app_name} is powered by Hydra.
25
+
26
+ '
27
+ footer: 'Powered by Hydra (https://hydra.cc)
28
+
29
+ Use --hydra-help to view Hydra specific help
30
+
31
+ '
32
+ template: '${hydra.help.header}
33
+
34
+ == Configuration groups ==
35
+
36
+ Compose your configuration from those groups (group=option)
37
+
38
+
39
+ $APP_CONFIG_GROUPS
40
+
41
+
42
+ == Config ==
43
+
44
+ Override anything in the config (foo.bar=value)
45
+
46
+
47
+ $CONFIG
48
+
49
+
50
+ ${hydra.help.footer}
51
+
52
+ '
53
+ hydra_help:
54
+ template: 'Hydra (${hydra.runtime.version})
55
+
56
+ See https://hydra.cc for more info.
57
+
58
+
59
+ == Flags ==
60
+
61
+ $FLAGS_HELP
62
+
63
+
64
+ == Configuration groups ==
65
+
66
+ Compose your configuration from those groups (For example, append hydra/job_logging=disabled
67
+ to command line)
68
+
69
+
70
+ $HYDRA_CONFIG_GROUPS
71
+
72
+
73
+ Use ''--cfg hydra'' to Show the Hydra config.
74
+
75
+ '
76
+ hydra_help: ???
77
+ hydra_logging:
78
+ version: 1
79
+ formatters:
80
+ simple:
81
+ format: '[%(asctime)s][HYDRA] %(message)s'
82
+ handlers:
83
+ console:
84
+ class: logging.StreamHandler
85
+ formatter: simple
86
+ stream: ext://sys.stdout
87
+ root:
88
+ level: INFO
89
+ handlers:
90
+ - console
91
+ loggers:
92
+ logging_example:
93
+ level: DEBUG
94
+ disable_existing_loggers: false
95
+ job_logging:
96
+ version: 1
97
+ formatters:
98
+ simple:
99
+ format: '[%(asctime)s][%(name)s][%(levelname)s] - %(message)s'
100
+ handlers:
101
+ console:
102
+ class: logging.StreamHandler
103
+ formatter: simple
104
+ stream: ext://sys.stdout
105
+ file:
106
+ class: logging.FileHandler
107
+ formatter: simple
108
+ filename: ${hydra.runtime.output_dir}/${hydra.job.name}.log
109
+ root:
110
+ level: INFO
111
+ handlers:
112
+ - console
113
+ - file
114
+ disable_existing_loggers: false
115
+ env: {}
116
+ mode: RUN
117
+ searchpath: []
118
+ callbacks: {}
119
+ output_subdir: .hydra
120
+ overrides:
121
+ hydra:
122
+ - hydra.mode=RUN
123
+ task:
124
+ - agent=diffusion
125
+ - suite=frankagym
126
+ - suite/frankagym_task@_global_=insertion
127
+ job:
128
+ name: eval_policy
129
+ chdir: true
130
+ override_dirname: agent=diffusion,suite/frankagym_task@_global_=insertion,suite=frankagym
131
+ id: ???
132
+ num: ???
133
+ config_name: config_eval
134
+ env_set: {}
135
+ env_copy: []
136
+ config:
137
+ override_dirname:
138
+ kv_sep: '='
139
+ item_sep: ','
140
+ exclude_keys: []
141
+ runtime:
142
+ version: 1.3.2
143
+ version_base: '1.1'
144
+ cwd: /home/leonmkim/fish_leon/FISH
145
+ config_sources:
146
+ - path: hydra.conf
147
+ schema: pkg
148
+ provider: hydra
149
+ - path: /home/leonmkim/fish_leon/FISH/cfgs
150
+ schema: file
151
+ provider: main
152
+ - path: ''
153
+ schema: structured
154
+ provider: schema
155
+ output_dir: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/204954
156
+ choices:
157
+ suite: frankagym
158
+ suite/frankagym_task@_global_: insertion
159
+ agent: diffusion
160
+ hydra/env: default
161
+ hydra/callbacks: null
162
+ hydra/job_logging: default
163
+ hydra/hydra_logging: default
164
+ hydra/hydra_help: default
165
+ hydra/help: default
166
+ hydra/sweeper: basic
167
+ hydra/launcher: submitit_local
168
+ hydra/output: default
169
+ verbose: false
204954/.hydra/overrides.yaml ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ - agent=diffusion
2
+ - suite=frankagym
3
+ - suite/frankagym_task@_global_=insertion
204954/eval_policy.log ADDED
@@ -0,0 +1,15 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ [2024-12-10 20:49:54,427][py.warnings][WARNING] - /home/leonmkim/fish_leon/FISH/eval_policy.py:428: UserWarning:
2
+ The version_base parameter is not specified.
3
+ Please specify a compatability version level, or None.
4
+ Will assume defaults for version 1.1
5
+ @hydra.main(config_path='cfgs', config_name='config_eval')
6
+
7
+ [2024-12-10 20:49:54,430][py.warnings][WARNING] - /home/leonmkim/fish_leon/FISH/eval_policy.py:365: UserWarning:
8
+ The version_base parameter is not specified.
9
+ Please specify a compatability version level, or None.
10
+ Will assume defaults for version 1.1
11
+ hydra.initialize(
12
+
13
+ [2024-12-10 20:49:55,731][py.warnings][WARNING] - /home/leonmkim/fish_leon/FISH/eval_policy.py:414: FutureWarning: You are using `torch.load` with `weights_only=False` (the current default value), which uses the default pickle module implicitly. It is possible to construct malicious pickle data which will execute arbitrary code during unpickling (See https://github.com/pytorch/pytorch/blob/main/SECURITY.md#untrusted-models for more details). In a future release, the default value for `weights_only` will be flipped to `True`. This limits the functions that could be executed during unpickling. Arbitrary objects will no longer be allowed to be loaded via this mode unless they are explicitly allowlisted by the user via `torch.serialization.add_safe_globals`. We recommend you start setting `weights_only=True` for any use case where you don't have full control of the loaded file. Please open an issue on GitHub for any issues related to this experimental feature.
14
+ payload = torch.load(f)
15
+
205123/.hydra/config.yaml ADDED
@@ -0,0 +1,349 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ root_dir: /home/${oc.env:USER}/fish_leon
2
+ nstep: 3
3
+ seed: 41
4
+ dataset_shuffle_seed: ${seed}
5
+ device: cuda
6
+ save_video: true
7
+ save_buffer: true
8
+ use_tb: true
9
+ baseline: false
10
+ use_wandb: true
11
+ eval: true
12
+ process_contact_features: ${eval}
13
+ obs_type: pixels
14
+ use_color: true
15
+ use_depth: true
16
+ use_masks: false
17
+ mask_list:
18
+ - EE_obj_mask
19
+ mask_representation: channels
20
+ crop_hw:
21
+ - 144
22
+ - 144
23
+ crop_down_offset: 48
24
+ color_crop_type: null
25
+ depth_crop_type: null
26
+ segmask_crop_type: null
27
+ add_crop_binary_mask: false
28
+ add_coord_conv_map: false
29
+ use_context_color: false
30
+ use_context_depth: false
31
+ use_context_segmask: false
32
+ context_color_crop_type: null
33
+ context_depth_crop_type: null
34
+ context_segmask_crop_type: null
35
+ context_add_crop_binary_mask: false
36
+ context_add_coord_conv_map: false
37
+ use_contact_map: false
38
+ use_sdf_maps: false
39
+ use_normals_maps: false
40
+ which_objects: both
41
+ max_contact_prob: 0.1
42
+ max_depth: 2.0
43
+ grasped_dtc_max_value: 0.105
44
+ env_dtc_max_value: 0.425
45
+ grasped_normals_mask_max_dtc_value: 0.105
46
+ env_normals_mask_max_dtc_value: 0.425
47
+ clamp_dtc: true
48
+ dtc_adaptive_normalization: false
49
+ mask_normals_within_sdf: true
50
+ adaptive_normals_mask: true
51
+ learnable_contact_preprocess_params: false
52
+ contact_model_name: local_multitask_outhd64all_home_crop_h144w144d48_ctxt_seed_183386_epoch_9
53
+ contact_estimation_model_ckpt_path: ~/fish_leon/contact_estimation/artifacts/175604_2/checkpoints/epoch=09-val_loss=0.00.ckpt
54
+ num_eval: 5
55
+ debug_timestamps: false
56
+ open_loop: false
57
+ action_trajectories: true
58
+ stop_after_action: false
59
+ interpolation_frequency: 25
60
+ policy_frequency: 5
61
+ wait_for_new_camera_frames: true
62
+ random_start: false
63
+ eval_starts: ${root_dir}/FISH/eval_starts/${suite.name}_${obs_type}/${task_name}
64
+ train_demo_idxs_list_or_num: null
65
+ num_valid_demos: null
66
+ val_num_groups: 3
67
+ name_of_expert_demo: 192_240x320_multiobj_twodim_recovery_5hz_zstd7_EE_pxl_coords_expert_demos_imp_act
68
+ expert_dataset_dirpath: ${root_dir}/FISH/expert_demos/${suite.name}/${task_name}/${name_of_expert_demo}
69
+ expert_dataset: ${expert_dataset_dirpath}/demos.zarr
70
+ action_key: ${oc.if_else:${action_trajectories}, 'action_trajectory_${interpolation_frequency}hz',
71
+ 'action'}
72
+ semantic_demo_grouping_name: semantic_demo_grouping.yaml
73
+ semantic_demo_grouping: ${expert_dataset_dirpath}/${semantic_demo_grouping_name}
74
+ expert_dataset_config: ${expert_dataset_dirpath}/demo_config.yaml
75
+ bc_regularize: false
76
+ bc_weight_type: qfilter
77
+ load_checkpoint: ${agent.load_checkpoint}
78
+ wandb_run_id: '265626'
79
+ true_action_history: true
80
+ wandb_notes: null
81
+ checkpoint_epoch: 12000
82
+ load_residual_weight: false
83
+ checkpoint_root_dir: /home/${oc.env:USER}/fish_leon/FISH
84
+ checkpoint_weight_dir: ${checkpoint_root_dir}/exp_local/${suite.name}_${obs_type}/${task_name}/${wandb_run_id}
85
+ residual_weight: ${root_dir}/FISH/weights/${suite.name}_${obs_type}/${task_name}/weight.pt
86
+ experiment_dir: ./exp_local/${suite.name}_${obs_type}/${task_name}/${wandb_run_id}
87
+ agent:
88
+ _target_: agent.diffusion_policy.DiffusionPolicyAgent
89
+ name: diffusion_policy
90
+ load_checkpoint: ${eval}
91
+ device: ${device}
92
+ n_obs_steps: ${.config.policy_cfg.n_obs_steps}
93
+ suite_name: ${suite.name}
94
+ obs_type: ${obs_type}
95
+ enable_arm: ${eval}
96
+ enable_camera: ${eval}
97
+ use_tb: ${use_tb}
98
+ desired_image_shape:
99
+ - 13
100
+ - 180
101
+ - 240
102
+ orig_cam_shape:
103
+ - 3
104
+ - 240
105
+ - 320
106
+ config:
107
+ _target_: agent.diffusion_policy.DiffusionPolicyAgentConfig
108
+ compile: false
109
+ device: ${device}
110
+ cam_resize_shape: ${agent.desired_image_shape}
111
+ orig_cam_shape: ${agent.orig_cam_shape}
112
+ policy_frequency: ${policy_frequency}
113
+ interpolation_frequency: ${interpolation_frequency}
114
+ policy_cfg:
115
+ _target_: lerobot.common.policies.diffusion.configuration_diffusion.DiffusionConfig
116
+ n_obs_steps: 1
117
+ horizon: 36
118
+ n_action_steps: ${agent.config.policy_cfg.horizon}
119
+ input_shapes:
120
+ observation.image: ${agent.config.cam_resize_shape}
121
+ context_observation.image: ${agent.config.cam_resize_shape}
122
+ observation.state:
123
+ - 8
124
+ observation.action_history:
125
+ - 7
126
+ output_shapes:
127
+ action:
128
+ - 7
129
+ input_normalization_modes:
130
+ observation.image: mean_std
131
+ observation.state: min_max
132
+ observation.action_history: min_max
133
+ output_normalization_modes:
134
+ action: min_max
135
+ vision_backbone: resnet18
136
+ pretrained_backbone_weights: null
137
+ transforms:
138
+ - _target_: torchaug.transforms.RandomAffine
139
+ degrees:
140
+ - -5
141
+ - 5
142
+ translate:
143
+ - 0.05
144
+ - 0.05
145
+ batch_transform: true
146
+ num_chunks: -1
147
+ batch_inplace: true
148
+ - _target_: torchaug.transforms.RandomColorJitter
149
+ brightness: 0.3
150
+ contrast: 0.4
151
+ saturation: 0.5
152
+ hue: 0.08
153
+ batch_transform: true
154
+ num_chunks: -1
155
+ batch_inplace: true
156
+ use_group_norm: true
157
+ spatial_softmax_num_keypoints: 32
158
+ action_history_encoder_config:
159
+ _target_: lerobot.common.policies.diffusion.configuration_diffusion.Unet1dEncoderConfig
160
+ in_channels: 7
161
+ out_channels: 32
162
+ history_length: ${agent.config.policy_cfg.n_action_steps}
163
+ kernel_size: ${agent.config.policy_cfg.kernel_size}
164
+ downsample_kernel_size: 3
165
+ downsample_stride: 2
166
+ downsample_padding: 1
167
+ down_dims:
168
+ - 256
169
+ - 512
170
+ - 1024
171
+ kernel_size: 5
172
+ n_groups: 8
173
+ diffusion_step_embed_dim: 128
174
+ use_film_scale_modulation: true
175
+ noise_scheduler_type: DDIM
176
+ beta_schedule: squaredcos_cap_v2
177
+ beta_start: 0.0001
178
+ beta_end: 0.02
179
+ prediction_type: epsilon
180
+ clip_sample: true
181
+ clip_sample_range: 1.0
182
+ num_train_timesteps: 50
183
+ num_inference_steps: 10
184
+ do_mask_loss_for_padding: false
185
+ train_cfg:
186
+ _target_: utils.TrainConfig
187
+ lr: 0.0001
188
+ lr_scheduler: cosine
189
+ lr_warmup_steps: 500
190
+ adam_betas:
191
+ - 0.95
192
+ - 0.999
193
+ adam_eps: 1.0e-08
194
+ adam_weight_decay: 1.0e-06
195
+ grad_clip_norm: 10
196
+ offline_steps: ${num_train_frames_diffusion}
197
+ use_amp: true
198
+ observation_cfg:
199
+ _target_: agent.encoder.VisualFeatureSet
200
+ use_depth: ${use_depth}
201
+ use_color: ${use_color}
202
+ mask_input_dict:
203
+ _target_: agent.encoder.MaskInputDict
204
+ enable: ${use_masks}
205
+ representation: ${mask_representation}
206
+ mask_list: ${mask_list}
207
+ crop_input_config:
208
+ _target_: agent.encoder.CropInputConfig
209
+ color_crop_type: ${color_crop_type}
210
+ depth_crop_type: ${depth_crop_type}
211
+ segmask_crop_type: ${segmask_crop_type}
212
+ crop_hw: ${crop_hw}
213
+ crop_down_offset: ${crop_down_offset}
214
+ add_crop_binary_mask: ${add_crop_binary_mask}
215
+ add_coord_conv_map: ${add_coord_conv_map}
216
+ context_input_config:
217
+ _target_: agent.encoder.ContextInputConfig
218
+ use_color: ${use_context_color}
219
+ use_depth: ${use_context_depth}
220
+ mask_input_dict:
221
+ _target_: agent.encoder.MaskInputDict
222
+ enable: ${use_context_segmask}
223
+ representation: ${mask_representation}
224
+ mask_list: ${mask_list}
225
+ crop_input_config:
226
+ _target_: agent.encoder.CropInputConfig
227
+ color_crop_type: ${context_color_crop_type}
228
+ depth_crop_type: ${context_depth_crop_type}
229
+ segmask_crop_type: ${context_segmask_crop_type}
230
+ crop_hw: ${crop_hw}
231
+ crop_down_offset: ${crop_down_offset}
232
+ add_crop_binary_mask: ${context_add_crop_binary_mask}
233
+ add_coord_conv_map: ${context_add_coord_conv_map}
234
+ mask_soft_approx_scheduler_config:
235
+ _target_: agent.encoder.MaskSoftApproxSchedulerConfig
236
+ num_steps: 40000
237
+ initial_value: 10.0
238
+ final_value: 1000.0
239
+ interpolation_scheme: constant
240
+ use_contact_map: ${use_contact_map}
241
+ use_sdf_maps: ${use_sdf_maps}
242
+ use_normals_maps: ${use_normals_maps}
243
+ which_objects: ${which_objects}
244
+ grasped_dtc_max_value: ${grasped_dtc_max_value}
245
+ env_dtc_max_value: ${env_dtc_max_value}
246
+ grasped_normals_mask_max_dtc_value: ${grasped_normals_mask_max_dtc_value}
247
+ env_normals_mask_max_dtc_value: ${env_normals_mask_max_dtc_value}
248
+ clamp_dtc: ${clamp_dtc}
249
+ max_contact_prob: ${max_contact_prob}
250
+ mask_normals_within_sdf: ${mask_normals_within_sdf}
251
+ dtc_adaptive_normalization: ${dtc_adaptive_normalization}
252
+ adaptive_normals_mask: ${adaptive_normals_mask}
253
+ max_depth: ${max_depth}
254
+ image_shape: ${agent.desired_image_shape}
255
+ learnable_contact_preprocess_params: ${learnable_contact_preprocess_params}
256
+ learning_rate: ${agent.config.train_cfg.lr}
257
+ weight_decay: 0.0
258
+ contact_model_name: ${contact_model_name}
259
+ zero_centered: false
260
+ suite:
261
+ suite: frankagym
262
+ name: frankagym
263
+ frame_stack: ${agent.n_obs_steps}
264
+ action_repeat: 1
265
+ discount: 0.99
266
+ hidden_dim: 1024
267
+ num_train_frames: 2010
268
+ num_seed_frames: 260
269
+ num_train_epochs: 5000
270
+ validate_every_epochs: 100
271
+ validate_diffusion_on_action_loss_every_epochs: 500
272
+ train_eval_diffusion_on_action_loss_every_epochs: 500
273
+ check_topk_every_epochs: 10
274
+ save_snapshot_every_epochs: 5000
275
+ eval_every_frames: 2000
276
+ num_eval_episodes: 5
277
+ save_snapshot: true
278
+ wait_for_user_to_start_episode: true
279
+ task_make_fn:
280
+ _target_: suite.frankagym.make
281
+ name: ${task_name}
282
+ height: 240
283
+ width: 320
284
+ frame_stack: ${suite.frame_stack}
285
+ action_repeat: ${suite.action_repeat}
286
+ seed: ${seed}
287
+ enable_arm: ${agent.enable_arm}
288
+ enable_gripper: ${enable_gripper}
289
+ start_with_gripper_open: ${start_with_gripper_open}
290
+ enable_camera: ${agent.enable_camera}
291
+ path_to_depth_extrinsics: ${path_to_depth_extrinsics}
292
+ contact_estimation_model_ckpt_path: ${contact_estimation_model_ckpt_path}
293
+ x_limit: ${x_limit}
294
+ y_limit: ${y_limit}
295
+ z_limit: ${z_limit}
296
+ device: ${device}
297
+ interpolation_frequency: ${interpolation_frequency}
298
+ policy_frequency: ${policy_frequency}
299
+ debug_timestamps: ${debug_timestamps}
300
+ stop_after_action: ${stop_after_action}
301
+ open_loop: ${open_loop}
302
+ wait_for_new_camera_frames: ${wait_for_new_camera_frames}
303
+ action_key: ${action_key}
304
+ action_trajectory_horizon: ${agent.config.policy_cfg.horizon}
305
+ action_trajectories: ${action_trajectories}
306
+ path_to_zarr_dataset: ${expert_dataset}
307
+ agent_policy_cfg: ???
308
+ true_action_history: ${true_action_history}
309
+ num_train_frames_bc: 50000
310
+ num_train_frames_drq: 1100000
311
+ stddev_schedule_drq: linear(1.0,0.1,100000)
312
+ task_name: FrankaInsertion-v1
313
+ num_train_frames_vinn: 25000
314
+ num_train_frames_diffusion: 1000000
315
+ num_train_epochs_bc: 5000
316
+ num_train_epochs_diffusion: 5000
317
+ validate_every_epochs_bc: 5
318
+ validate_every_epochs_diffusion: 25
319
+ validate_diffusion_on_action_loss_every_epochs: 50
320
+ train_eval_diffusion_on_action_loss_every_epochs: 500
321
+ check_topk_every_epochs: 5
322
+ check_topk_every_epochs_diffusion: ${validate_diffusion_on_action_loss_every_epochs}
323
+ save_snapshot_every_epochs_diffusion: 5000
324
+ x_limit:
325
+ - 0.2
326
+ - 0.7
327
+ y_limit:
328
+ - -0.4
329
+ - 0.4
330
+ z_limit:
331
+ - -0.05
332
+ - 0.55
333
+ home_displacement:
334
+ - 0.55
335
+ - 0.0
336
+ - 0.55
337
+ - 180.0
338
+ - 0.0
339
+ - 0.0
340
+ enable_gripper: true
341
+ start_with_gripper_open: true
342
+ offset_mask:
343
+ - 1
344
+ - 1
345
+ - 1
346
+ - 1
347
+ - 1
348
+ - 1
349
+ path_to_depth_extrinsics: ~/fish_leon/FISH/cfgs/camera_poses/camera_poses_L515/20240904-122305/color_tf_world.npy
205123/.hydra/hydra.yaml ADDED
@@ -0,0 +1,169 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ hydra:
2
+ run:
3
+ dir: ${experiment_dir}/${now:%H%M%S}
4
+ sweep:
5
+ dir: ${experiment_dir}/${now:%H%M%S}
6
+ subdir: ${hydra.job.num}
7
+ launcher:
8
+ submitit_folder: ${experiment_dir}/${now:%H%M%S}/.slurm
9
+ timeout_min: 60
10
+ cpus_per_task: null
11
+ gpus_per_node: null
12
+ tasks_per_node: 1
13
+ mem_gb: null
14
+ nodes: 1
15
+ name: ${hydra.job.name}
16
+ stderr_to_stdout: false
17
+ _target_: hydra_plugins.hydra_submitit_launcher.submitit_launcher.LocalLauncher
18
+ sweeper:
19
+ _target_: hydra._internal.core_plugins.basic_sweeper.BasicSweeper
20
+ max_batch_size: null
21
+ params: null
22
+ help:
23
+ app_name: ${hydra.job.name}
24
+ header: '${hydra.help.app_name} is powered by Hydra.
25
+
26
+ '
27
+ footer: 'Powered by Hydra (https://hydra.cc)
28
+
29
+ Use --hydra-help to view Hydra specific help
30
+
31
+ '
32
+ template: '${hydra.help.header}
33
+
34
+ == Configuration groups ==
35
+
36
+ Compose your configuration from those groups (group=option)
37
+
38
+
39
+ $APP_CONFIG_GROUPS
40
+
41
+
42
+ == Config ==
43
+
44
+ Override anything in the config (foo.bar=value)
45
+
46
+
47
+ $CONFIG
48
+
49
+
50
+ ${hydra.help.footer}
51
+
52
+ '
53
+ hydra_help:
54
+ template: 'Hydra (${hydra.runtime.version})
55
+
56
+ See https://hydra.cc for more info.
57
+
58
+
59
+ == Flags ==
60
+
61
+ $FLAGS_HELP
62
+
63
+
64
+ == Configuration groups ==
65
+
66
+ Compose your configuration from those groups (For example, append hydra/job_logging=disabled
67
+ to command line)
68
+
69
+
70
+ $HYDRA_CONFIG_GROUPS
71
+
72
+
73
+ Use ''--cfg hydra'' to Show the Hydra config.
74
+
75
+ '
76
+ hydra_help: ???
77
+ hydra_logging:
78
+ version: 1
79
+ formatters:
80
+ simple:
81
+ format: '[%(asctime)s][HYDRA] %(message)s'
82
+ handlers:
83
+ console:
84
+ class: logging.StreamHandler
85
+ formatter: simple
86
+ stream: ext://sys.stdout
87
+ root:
88
+ level: INFO
89
+ handlers:
90
+ - console
91
+ loggers:
92
+ logging_example:
93
+ level: DEBUG
94
+ disable_existing_loggers: false
95
+ job_logging:
96
+ version: 1
97
+ formatters:
98
+ simple:
99
+ format: '[%(asctime)s][%(name)s][%(levelname)s] - %(message)s'
100
+ handlers:
101
+ console:
102
+ class: logging.StreamHandler
103
+ formatter: simple
104
+ stream: ext://sys.stdout
105
+ file:
106
+ class: logging.FileHandler
107
+ formatter: simple
108
+ filename: ${hydra.runtime.output_dir}/${hydra.job.name}.log
109
+ root:
110
+ level: INFO
111
+ handlers:
112
+ - console
113
+ - file
114
+ disable_existing_loggers: false
115
+ env: {}
116
+ mode: RUN
117
+ searchpath: []
118
+ callbacks: {}
119
+ output_subdir: .hydra
120
+ overrides:
121
+ hydra:
122
+ - hydra.mode=RUN
123
+ task:
124
+ - agent=diffusion
125
+ - suite=frankagym
126
+ - suite/frankagym_task@_global_=insertion
127
+ job:
128
+ name: eval_robot
129
+ chdir: true
130
+ override_dirname: agent=diffusion,suite/frankagym_task@_global_=insertion,suite=frankagym
131
+ id: ???
132
+ num: ???
133
+ config_name: config_eval
134
+ env_set: {}
135
+ env_copy: []
136
+ config:
137
+ override_dirname:
138
+ kv_sep: '='
139
+ item_sep: ','
140
+ exclude_keys: []
141
+ runtime:
142
+ version: 1.3.2
143
+ version_base: '1.1'
144
+ cwd: /home/leonmkim/fish_leon/FISH
145
+ config_sources:
146
+ - path: hydra.conf
147
+ schema: pkg
148
+ provider: hydra
149
+ - path: /home/leonmkim/fish_leon/FISH/cfgs
150
+ schema: file
151
+ provider: main
152
+ - path: ''
153
+ schema: structured
154
+ provider: schema
155
+ output_dir: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123
156
+ choices:
157
+ suite: frankagym
158
+ suite/frankagym_task@_global_: insertion
159
+ agent: diffusion
160
+ hydra/env: default
161
+ hydra/callbacks: null
162
+ hydra/job_logging: default
163
+ hydra/hydra_logging: default
164
+ hydra/hydra_help: default
165
+ hydra/help: default
166
+ hydra/sweeper: basic
167
+ hydra/launcher: submitit_local
168
+ hydra/output: default
169
+ verbose: false
205123/.hydra/overrides.yaml ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ - agent=diffusion
2
+ - suite=frankagym
3
+ - suite/frankagym_task@_global_=insertion
205123/episode_rosbags/aligned_depth_to_color_K.npy ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4a962c703d20da282f5e009d432dff51df4ebd22f3386699b6754ea9cfc06a55
3
+ size 200
205123/episode_rosbags/cam_tf_world.npy ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:29313ba240dd65ebbc79056d5582a97c9586cf2a1d4a1e0db13b87b49152cc9e
3
+ size 256
205123/episode_rosbags/color_K.npy ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4a962c703d20da282f5e009d432dff51df4ebd22f3386699b6754ea9cfc06a55
3
+ size 200
205123/episode_rosbags/depth_K.npy ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:cc601d2fecd31c5513c76a88b5d4d8059adc1f92dad682e2d755d89a66d8fdf7
3
+ size 200
205123/eval_robot.log ADDED
@@ -0,0 +1,12 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ [2024-12-10 20:51:23,588][py.warnings][WARNING] - /home/leonmkim/fish_leon/FISH/eval_robot.py:503: UserWarning:
2
+ The version_base parameter is not specified.
3
+ Please specify a compatability version level, or None.
4
+ Will assume defaults for version 1.1
5
+ @hydra.main(config_path='cfgs', config_name='config_eval')
6
+
7
+ [2024-12-10 20:51:23,591][py.warnings][WARNING] - /home/leonmkim/fish_leon/FISH/eval_robot.py:439: UserWarning:
8
+ The version_base parameter is not specified.
9
+ Please specify a compatability version level, or None.
10
+ Will assume defaults for version 1.1
11
+ hydra.initialize(
12
+
205123/eval_video/0_eval.mp4 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b6b0b3ba17b7513a5df9bca8084b0eda15d2f82df085783449312a47400bba10
3
+ size 942048
205123/tb/events.out.tfevents.1733881891.leonmkim-ROG-Strix-G15CS-G15CS.596892.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c19a2cb159a40df6149928fa963c721dd4fb49916ec660df1f61fffc949c49ec
3
+ size 88
205123/wandb/debug-internal.log ADDED
@@ -0,0 +1,351 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ 2024-12-10 20:51:30,311 INFO StreamThr :597014 [internal.py:wandb_internal():85] W&B internal server running at pid: 597014, started at: 2024-12-10 20:51:30.311157
2
+ 2024-12-10 20:51:30,312 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: status
3
+ 2024-12-10 20:51:30,325 INFO WriterThread:597014 [datastore.py:open_for_write():87] open: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/wandb/run-20241210_205130-svysfc56/run-svysfc56.wandb
4
+ 2024-12-10 20:51:30,325 DEBUG SenderThread:597014 [sender.py:send():379] send: header
5
+ 2024-12-10 20:51:30,336 DEBUG SenderThread:597014 [sender.py:send():379] send: run
6
+ 2024-12-10 20:51:30,645 INFO SenderThread:597014 [dir_watcher.py:__init__():211] watching files in: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/wandb/run-20241210_205130-svysfc56/files
7
+ 2024-12-10 20:51:30,645 INFO SenderThread:597014 [sender.py:_start_run_threads():1188] run started: svysfc56 with start time 1733881890.312299
8
+ 2024-12-10 20:51:30,656 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: check_version
9
+ 2024-12-10 20:51:30,656 DEBUG SenderThread:597014 [sender.py:send_request():406] send_request: check_version
10
+ 2024-12-10 20:51:30,743 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: run_start
11
+ 2024-12-10 20:51:30,748 DEBUG HandlerThread:597014 [system_info.py:__init__():26] System info init
12
+ 2024-12-10 20:51:30,748 DEBUG HandlerThread:597014 [system_info.py:__init__():41] System info init done
13
+ 2024-12-10 20:51:30,748 INFO HandlerThread:597014 [system_monitor.py:start():194] Starting system monitor
14
+ 2024-12-10 20:51:30,748 INFO SystemMonitor:597014 [system_monitor.py:_start():158] Starting system asset monitoring threads
15
+ 2024-12-10 20:51:30,749 INFO HandlerThread:597014 [system_monitor.py:probe():214] Collecting system info
16
+ 2024-12-10 20:51:30,749 INFO SystemMonitor:597014 [interfaces.py:start():188] Started cpu monitoring
17
+ 2024-12-10 20:51:30,750 INFO SystemMonitor:597014 [interfaces.py:start():188] Started disk monitoring
18
+ 2024-12-10 20:51:30,750 INFO SystemMonitor:597014 [interfaces.py:start():188] Started gpu monitoring
19
+ 2024-12-10 20:51:30,751 INFO SystemMonitor:597014 [interfaces.py:start():188] Started memory monitoring
20
+ 2024-12-10 20:51:30,751 INFO SystemMonitor:597014 [interfaces.py:start():188] Started network monitoring
21
+ 2024-12-10 20:51:30,761 DEBUG HandlerThread:597014 [system_info.py:probe():152] Probing system
22
+ 2024-12-10 20:51:30,762 DEBUG HandlerThread:597014 [system_info.py:_probe_git():137] Probing git
23
+ 2024-12-10 20:51:30,767 DEBUG HandlerThread:597014 [system_info.py:_probe_git():145] Probing git done
24
+ 2024-12-10 20:51:30,767 DEBUG HandlerThread:597014 [system_info.py:probe():200] Probing system done
25
+ 2024-12-10 20:51:30,767 DEBUG HandlerThread:597014 [system_monitor.py:probe():223] {'os': 'Linux-5.15.0-125-generic-x86_64-with-glibc2.31', 'python': '3.10.14', 'heartbeatAt': '2024-12-11T01:51:30.761221', 'startedAt': '2024-12-11T01:51:30.302896', 'docker': None, 'cuda': None, 'args': ('agent=diffusion', 'suite=frankagym', 'suite/frankagym_task@_global_=insertion'), 'state': 'running', 'program': '/home/leonmkim/fish_leon/FISH/eval_robot.py', 'codePathLocal': None, 'codePath': 'FISH/eval_robot.py', 'git': {'remote': 'https://github.com/leonmkim/fish_leon.git', 'commit': '4f7c0421eb84a7caf6bf5670f68777523362676f'}, 'email': 'leonmkim@seas.upenn.edu', 'root': '/home/leonmkim/fish_leon', 'host': 'leonmkim-ROG-Strix-G15CS-G15CS', 'username': 'leonmkim', 'executable': '/home/leonmkim/.pyenv/versions/lerobot/bin/python', 'cpu_count': 8, 'cpu_count_logical': 8, 'cpu_freq': {'current': 4124.7015, 'min': 800.0, 'max': 4700.0}, 'cpu_freq_per_core': [{'current': 3000.0, 'min': 800.0, 'max': 4700.0}, {'current': 4499.948, 'min': 800.0, 'max': 4700.0}, {'current': 4499.997, 'min': 800.0, 'max': 4700.0}, {'current': 3000.0, 'min': 800.0, 'max': 4700.0}, {'current': 3000.0, 'min': 800.0, 'max': 4700.0}, {'current': 3000.0, 'min': 800.0, 'max': 4700.0}, {'current': 3000.0, 'min': 800.0, 'max': 4700.0}, {'current': 4499.999, 'min': 800.0, 'max': 4700.0}], 'disk': {'/': {'total': 915.3232879638672, 'used': 364.1780319213867}}, 'gpu': 'NVIDIA GeForce RTX 2070 SUPER', 'gpu_count': 1, 'gpu_devices': [{'name': 'NVIDIA GeForce RTX 2070 SUPER', 'memory_total': 8589934592}], 'memory': {'total': 62.71596908569336}}
26
+ 2024-12-10 20:51:30,768 INFO HandlerThread:597014 [system_monitor.py:probe():224] Finished collecting system info
27
+ 2024-12-10 20:51:30,768 INFO HandlerThread:597014 [system_monitor.py:probe():227] Publishing system info
28
+ 2024-12-10 20:51:30,768 DEBUG HandlerThread:597014 [system_info.py:_save_code():44] Saving code
29
+ 2024-12-10 20:51:30,769 DEBUG HandlerThread:597014 [system_info.py:_save_code():67] Saving code done
30
+ 2024-12-10 20:51:30,769 DEBUG HandlerThread:597014 [system_info.py:_save_patches():84] Saving git patches
31
+ 2024-12-10 20:51:30,832 DEBUG HandlerThread:597014 [system_info.py:_save_patches():126] Saving git patches done
32
+ 2024-12-10 20:51:30,832 INFO HandlerThread:597014 [system_monitor.py:probe():229] Finished publishing system info
33
+ 2024-12-10 20:51:30,836 DEBUG SenderThread:597014 [sender.py:send():379] send: files
34
+ 2024-12-10 20:51:30,836 INFO SenderThread:597014 [sender.py:_save_file():1454] saving file wandb-metadata.json with policy now
35
+ 2024-12-10 20:51:30,836 INFO SenderThread:597014 [sender.py:_save_file():1454] saving file code/FISH/eval_robot.py with policy now
36
+ 2024-12-10 20:51:30,836 INFO SenderThread:597014 [sender.py:_save_file():1454] saving file diff.patch with policy now
37
+ 2024-12-10 20:51:31,124 INFO wandb-upload_2:597014 [upload_job.py:push():130] Uploaded file /tmp/tmp01kl0ofywandb/tiz5ae9w-diff.patch
38
+ 2024-12-10 20:51:31,126 INFO wandb-upload_0:597014 [upload_job.py:push():130] Uploaded file /tmp/tmp01kl0ofywandb/p6o167mp-wandb-metadata.json
39
+ 2024-12-10 20:51:31,143 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: python_packages
40
+ 2024-12-10 20:51:31,144 DEBUG SenderThread:597014 [sender.py:send_request():406] send_request: python_packages
41
+ 2024-12-10 20:51:31,145 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: stop_status
42
+ 2024-12-10 20:51:31,145 DEBUG SenderThread:597014 [sender.py:send_request():406] send_request: stop_status
43
+ 2024-12-10 20:51:31,146 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
44
+ 2024-12-10 20:51:31,147 INFO HandlerThread:597014 [handler.py:handle_tbrecord():783] handling tbrecord: tbrecord {
45
+ log_dir: "/home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/tb"
46
+ save: true
47
+ }
48
+ control {
49
+ relay_id: "42f7013eb6ef4a01b21dd778abc8c686"
50
+ }
51
+ _info {
52
+ stream_id: "svysfc56"
53
+ }
54
+
55
+ 2024-12-10 20:51:31,147 DEBUG HandlerThread:597014 [config_util.py:dict_from_config_file():70] no default config file found in config-defaults.yaml
56
+ 2024-12-10 20:51:31,173 INFO wandb-upload_1:597014 [upload_job.py:push():130] Uploaded file /tmp/tmp01kl0ofywandb/9ln43j5e-code/FISH/eval_robot.py
57
+ 2024-12-10 20:51:31,218 DEBUG SenderThread:597014 [sender.py:send():379] send: telemetry
58
+ 2024-12-10 20:51:31,218 DEBUG SenderThread:597014 [sender.py:send():379] send: tbrecord
59
+ 2024-12-10 20:51:31,218 DEBUG SenderThread:597014 [sender.py:send():379] send: files
60
+ 2024-12-10 20:51:31,218 INFO SenderThread:597014 [sender.py:_save_file():1454] saving file events.out.tfevents.1733881891.leonmkim-ROG-Strix-G15CS-G15CS.596892.0 with policy live
61
+ 2024-12-10 20:51:31,415 INFO wandb-upload_2:597014 [upload_job.py:push():130] Uploaded file /tmp/tmp01kl0ofywandb/08emxf8k-events.out.tfevents.1733881891.leonmkim-ROG-Strix-G15CS-G15CS.596892.0
62
+ 2024-12-10 20:51:31,647 INFO Thread-12 :597014 [dir_watcher.py:_on_file_created():271] file/dir created: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/wandb/run-20241210_205130-svysfc56/files/diff.patch
63
+ 2024-12-10 20:51:31,647 INFO Thread-12 :597014 [dir_watcher.py:_on_file_created():271] file/dir created: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/wandb/run-20241210_205130-svysfc56/files/code/FISH/eval_robot.py
64
+ 2024-12-10 20:51:31,647 INFO Thread-12 :597014 [dir_watcher.py:_on_file_created():271] file/dir created: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/wandb/run-20241210_205130-svysfc56/files/requirements.txt
65
+ 2024-12-10 20:51:31,647 INFO Thread-12 :597014 [dir_watcher.py:_on_file_created():271] file/dir created: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/wandb/run-20241210_205130-svysfc56/files/output.log
66
+ 2024-12-10 20:51:31,647 INFO Thread-12 :597014 [dir_watcher.py:_on_file_created():271] file/dir created: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/wandb/run-20241210_205130-svysfc56/files/wandb-metadata.json
67
+ 2024-12-10 20:51:31,647 INFO Thread-12 :597014 [dir_watcher.py:_on_file_created():271] file/dir created: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/wandb/run-20241210_205130-svysfc56/files/events.out.tfevents.1733881891.leonmkim-ROG-Strix-G15CS-G15CS.596892.0
68
+ 2024-12-10 20:51:31,647 INFO Thread-12 :597014 [dir_watcher.py:_on_file_created():271] file/dir created: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/wandb/run-20241210_205130-svysfc56/files/code/FISH
69
+ 2024-12-10 20:51:31,647 INFO Thread-12 :597014 [dir_watcher.py:_on_file_created():271] file/dir created: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/wandb/run-20241210_205130-svysfc56/files/code
70
+ 2024-12-10 20:51:32,144 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
71
+ 2024-12-10 20:51:33,145 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
72
+ 2024-12-10 20:51:33,646 INFO Thread-12 :597014 [dir_watcher.py:_on_file_modified():288] file/dir modified: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/wandb/run-20241210_205130-svysfc56/files/output.log
73
+ 2024-12-10 20:51:34,145 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
74
+ 2024-12-10 20:51:35,145 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
75
+ 2024-12-10 20:51:36,144 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: status_report
76
+ 2024-12-10 20:51:36,145 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
77
+ 2024-12-10 20:51:37,145 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
78
+ 2024-12-10 20:51:38,145 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
79
+ 2024-12-10 20:51:39,145 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
80
+ 2024-12-10 20:51:40,145 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
81
+ 2024-12-10 20:51:41,144 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: status_report
82
+ 2024-12-10 20:51:41,146 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
83
+ 2024-12-10 20:51:42,146 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
84
+ 2024-12-10 20:51:43,146 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
85
+ 2024-12-10 20:51:44,146 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
86
+ 2024-12-10 20:51:45,146 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
87
+ 2024-12-10 20:51:46,144 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: stop_status
88
+ 2024-12-10 20:51:46,144 DEBUG SenderThread:597014 [sender.py:send_request():406] send_request: stop_status
89
+ 2024-12-10 20:51:46,186 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
90
+ 2024-12-10 20:51:46,300 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: status_report
91
+ 2024-12-10 20:51:46,908 DEBUG SenderThread:597014 [sender.py:send():379] send: config
92
+ 2024-12-10 20:51:47,146 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
93
+ 2024-12-10 20:51:48,146 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
94
+ 2024-12-10 20:51:49,146 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
95
+ 2024-12-10 20:51:49,651 INFO Thread-12 :597014 [dir_watcher.py:_on_file_modified():288] file/dir modified: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/wandb/run-20241210_205130-svysfc56/files/output.log
96
+ 2024-12-10 20:51:50,146 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
97
+ 2024-12-10 20:51:51,147 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
98
+ 2024-12-10 20:51:51,335 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: status_report
99
+ 2024-12-10 20:51:52,147 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
100
+ 2024-12-10 20:51:53,147 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
101
+ 2024-12-10 20:51:53,652 INFO Thread-12 :597014 [dir_watcher.py:_on_file_modified():288] file/dir modified: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/wandb/run-20241210_205130-svysfc56/files/output.log
102
+ 2024-12-10 20:51:54,147 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
103
+ 2024-12-10 20:51:55,147 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
104
+ 2024-12-10 20:51:56,147 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
105
+ 2024-12-10 20:51:57,145 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: status_report
106
+ 2024-12-10 20:51:57,147 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
107
+ 2024-12-10 20:51:57,654 INFO Thread-12 :597014 [dir_watcher.py:_on_file_modified():288] file/dir modified: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/wandb/run-20241210_205130-svysfc56/files/output.log
108
+ 2024-12-10 20:51:58,148 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
109
+ 2024-12-10 20:51:59,148 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
110
+ 2024-12-10 20:52:00,148 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
111
+ 2024-12-10 20:52:01,144 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: stop_status
112
+ 2024-12-10 20:52:01,144 DEBUG SenderThread:597014 [sender.py:send_request():406] send_request: stop_status
113
+ 2024-12-10 20:52:01,189 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
114
+ 2024-12-10 20:52:02,148 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
115
+ 2024-12-10 20:52:02,300 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: status_report
116
+ 2024-12-10 20:52:02,655 INFO Thread-12 :597014 [dir_watcher.py:_on_file_modified():288] file/dir modified: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/wandb/run-20241210_205130-svysfc56/files/config.yaml
117
+ 2024-12-10 20:52:03,148 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
118
+ 2024-12-10 20:52:04,149 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
119
+ 2024-12-10 20:52:05,149 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
120
+ 2024-12-10 20:52:06,182 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
121
+ 2024-12-10 20:52:07,149 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
122
+ 2024-12-10 20:52:07,657 INFO Thread-12 :597014 [dir_watcher.py:_on_file_modified():288] file/dir modified: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/wandb/run-20241210_205130-svysfc56/files/output.log
123
+ 2024-12-10 20:52:08,149 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
124
+ 2024-12-10 20:52:08,194 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: status_report
125
+ 2024-12-10 20:52:09,149 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
126
+ 2024-12-10 20:52:09,657 INFO Thread-12 :597014 [dir_watcher.py:_on_file_modified():288] file/dir modified: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/wandb/run-20241210_205130-svysfc56/files/output.log
127
+ 2024-12-10 20:52:10,149 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
128
+ 2024-12-10 20:52:11,149 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
129
+ 2024-12-10 20:52:11,658 INFO Thread-12 :597014 [dir_watcher.py:_on_file_modified():288] file/dir modified: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/wandb/run-20241210_205130-svysfc56/files/output.log
130
+ 2024-12-10 20:52:12,149 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
131
+ 2024-12-10 20:52:13,149 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
132
+ 2024-12-10 20:52:13,219 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: status_report
133
+ 2024-12-10 20:52:13,659 INFO Thread-12 :597014 [dir_watcher.py:_on_file_modified():288] file/dir modified: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/wandb/run-20241210_205130-svysfc56/files/output.log
134
+ 2024-12-10 20:52:14,150 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
135
+ 2024-12-10 20:52:15,150 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
136
+ 2024-12-10 20:52:15,659 INFO Thread-12 :597014 [dir_watcher.py:_on_file_modified():288] file/dir modified: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/wandb/run-20241210_205130-svysfc56/files/output.log
137
+ 2024-12-10 20:52:16,144 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: stop_status
138
+ 2024-12-10 20:52:16,144 DEBUG SenderThread:597014 [sender.py:send_request():406] send_request: stop_status
139
+ 2024-12-10 20:52:16,150 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
140
+ 2024-12-10 20:52:17,150 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
141
+ 2024-12-10 20:52:17,660 INFO Thread-12 :597014 [dir_watcher.py:_on_file_modified():288] file/dir modified: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/wandb/run-20241210_205130-svysfc56/files/output.log
142
+ 2024-12-10 20:52:18,150 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
143
+ 2024-12-10 20:52:18,404 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: status_report
144
+ 2024-12-10 20:52:19,152 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
145
+ 2024-12-10 20:52:19,661 INFO Thread-12 :597014 [dir_watcher.py:_on_file_modified():288] file/dir modified: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/wandb/run-20241210_205130-svysfc56/files/output.log
146
+ 2024-12-10 20:52:20,152 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
147
+ 2024-12-10 20:52:21,152 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
148
+ 2024-12-10 20:52:21,661 INFO Thread-12 :597014 [dir_watcher.py:_on_file_modified():288] file/dir modified: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/wandb/run-20241210_205130-svysfc56/files/output.log
149
+ 2024-12-10 20:52:22,152 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
150
+ 2024-12-10 20:52:23,152 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
151
+ 2024-12-10 20:52:23,407 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: status_report
152
+ 2024-12-10 20:52:23,662 INFO Thread-12 :597014 [dir_watcher.py:_on_file_modified():288] file/dir modified: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/wandb/run-20241210_205130-svysfc56/files/output.log
153
+ 2024-12-10 20:52:24,152 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
154
+ 2024-12-10 20:52:25,153 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
155
+ 2024-12-10 20:52:25,662 INFO Thread-12 :597014 [dir_watcher.py:_on_file_modified():288] file/dir modified: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/wandb/run-20241210_205130-svysfc56/files/output.log
156
+ 2024-12-10 20:52:26,153 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
157
+ 2024-12-10 20:52:27,153 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
158
+ 2024-12-10 20:52:27,663 INFO Thread-12 :597014 [dir_watcher.py:_on_file_modified():288] file/dir modified: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/wandb/run-20241210_205130-svysfc56/files/output.log
159
+ 2024-12-10 20:52:28,153 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
160
+ 2024-12-10 20:52:28,607 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: status_report
161
+ 2024-12-10 20:52:29,153 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
162
+ 2024-12-10 20:52:29,664 INFO Thread-12 :597014 [dir_watcher.py:_on_file_modified():288] file/dir modified: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/wandb/run-20241210_205130-svysfc56/files/output.log
163
+ 2024-12-10 20:52:30,154 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
164
+ 2024-12-10 20:52:30,752 DEBUG SystemMonitor:597014 [system_monitor.py:_start():172] Starting system metrics aggregation loop
165
+ 2024-12-10 20:52:30,752 DEBUG SenderThread:597014 [sender.py:send():379] send: stats
166
+ 2024-12-10 20:52:31,144 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: stop_status
167
+ 2024-12-10 20:52:31,144 DEBUG SenderThread:597014 [sender.py:send_request():406] send_request: stop_status
168
+ 2024-12-10 20:52:31,154 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
169
+ 2024-12-10 20:52:31,666 INFO Thread-12 :597014 [dir_watcher.py:_on_file_modified():288] file/dir modified: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/wandb/run-20241210_205130-svysfc56/files/output.log
170
+ 2024-12-10 20:52:32,154 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
171
+ 2024-12-10 20:52:33,154 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
172
+ 2024-12-10 20:52:33,667 INFO Thread-12 :597014 [dir_watcher.py:_on_file_modified():288] file/dir modified: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/wandb/run-20241210_205130-svysfc56/files/output.log
173
+ 2024-12-10 20:52:33,908 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: status_report
174
+ 2024-12-10 20:52:34,154 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
175
+ 2024-12-10 20:52:34,667 INFO Thread-12 :597014 [dir_watcher.py:_on_file_modified():288] file/dir modified: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/wandb/run-20241210_205130-svysfc56/files/output.log
176
+ 2024-12-10 20:52:35,154 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
177
+ 2024-12-10 20:52:35,668 INFO Thread-12 :597014 [dir_watcher.py:_on_file_modified():288] file/dir modified: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/wandb/run-20241210_205130-svysfc56/files/output.log
178
+ 2024-12-10 20:52:36,154 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
179
+ 2024-12-10 20:52:37,155 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
180
+ 2024-12-10 20:52:38,155 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
181
+ 2024-12-10 20:52:38,668 INFO Thread-12 :597014 [dir_watcher.py:_on_file_modified():288] file/dir modified: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/wandb/run-20241210_205130-svysfc56/files/output.log
182
+ 2024-12-10 20:52:39,155 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
183
+ 2024-12-10 20:52:39,371 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: status_report
184
+ 2024-12-10 20:52:40,155 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
185
+ 2024-12-10 20:52:41,155 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
186
+ 2024-12-10 20:52:42,155 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
187
+ 2024-12-10 20:52:43,155 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
188
+ 2024-12-10 20:52:44,155 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
189
+ 2024-12-10 20:52:44,374 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: partial_history
190
+ 2024-12-10 20:52:44,375 DEBUG SenderThread:597014 [sender.py:send():379] send: files
191
+ 2024-12-10 20:52:44,375 INFO SenderThread:597014 [sender.py:_save_file():1454] saving file media/videos/eval/video_0_b6b0b3ba17b7513a5df9.mp4 with policy now
192
+ 2024-12-10 20:52:44,375 DEBUG SenderThread:597014 [sender.py:send():379] send: history
193
+ 2024-12-10 20:52:44,375 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: status_report
194
+ 2024-12-10 20:52:44,375 DEBUG SenderThread:597014 [sender.py:send_request():406] send_request: summary_record
195
+ 2024-12-10 20:52:44,377 INFO SenderThread:597014 [sender.py:_save_file():1454] saving file wandb-summary.json with policy end
196
+ 2024-12-10 20:52:44,377 DEBUG SenderThread:597014 [sender.py:send():379] send: exit
197
+ 2024-12-10 20:52:44,377 INFO SenderThread:597014 [sender.py:send_exit():586] handling exit code: 0
198
+ 2024-12-10 20:52:44,377 INFO SenderThread:597014 [sender.py:send_exit():588] handling runtime: 73
199
+ 2024-12-10 20:52:44,377 INFO SenderThread:597014 [sender.py:_save_file():1454] saving file wandb-summary.json with policy end
200
+ 2024-12-10 20:52:44,377 INFO SenderThread:597014 [sender.py:send_exit():594] send defer
201
+ 2024-12-10 20:52:44,379 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: defer
202
+ 2024-12-10 20:52:44,379 INFO HandlerThread:597014 [handler.py:handle_request_defer():184] handle defer: 0
203
+ 2024-12-10 20:52:44,380 DEBUG SenderThread:597014 [sender.py:send_request():406] send_request: defer
204
+ 2024-12-10 20:52:44,380 INFO SenderThread:597014 [sender.py:send_request_defer():610] handle sender defer: 0
205
+ 2024-12-10 20:52:44,380 INFO SenderThread:597014 [sender.py:transition_state():614] send defer: 1
206
+ 2024-12-10 20:52:44,380 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: defer
207
+ 2024-12-10 20:52:44,380 INFO HandlerThread:597014 [handler.py:handle_request_defer():184] handle defer: 1
208
+ 2024-12-10 20:52:44,380 DEBUG SenderThread:597014 [sender.py:send_request():406] send_request: defer
209
+ 2024-12-10 20:52:44,380 INFO SenderThread:597014 [sender.py:send_request_defer():610] handle sender defer: 1
210
+ 2024-12-10 20:52:44,380 INFO SenderThread:597014 [sender.py:transition_state():614] send defer: 2
211
+ 2024-12-10 20:52:44,380 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: defer
212
+ 2024-12-10 20:52:44,380 INFO HandlerThread:597014 [handler.py:handle_request_defer():184] handle defer: 2
213
+ 2024-12-10 20:52:44,380 INFO HandlerThread:597014 [system_monitor.py:finish():203] Stopping system monitor
214
+ 2024-12-10 20:52:44,381 DEBUG SystemMonitor:597014 [system_monitor.py:_start():179] Finished system metrics aggregation loop
215
+ 2024-12-10 20:52:44,381 DEBUG SystemMonitor:597014 [system_monitor.py:_start():183] Publishing last batch of metrics
216
+ 2024-12-10 20:52:44,381 INFO HandlerThread:597014 [interfaces.py:finish():200] Joined cpu monitor
217
+ 2024-12-10 20:52:44,382 INFO HandlerThread:597014 [interfaces.py:finish():200] Joined disk monitor
218
+ 2024-12-10 20:52:44,439 INFO HandlerThread:597014 [interfaces.py:finish():200] Joined gpu monitor
219
+ 2024-12-10 20:52:44,439 INFO HandlerThread:597014 [interfaces.py:finish():200] Joined memory monitor
220
+ 2024-12-10 20:52:44,439 INFO HandlerThread:597014 [interfaces.py:finish():200] Joined network monitor
221
+ 2024-12-10 20:52:44,440 DEBUG SenderThread:597014 [sender.py:send_request():406] send_request: defer
222
+ 2024-12-10 20:52:44,440 INFO SenderThread:597014 [sender.py:send_request_defer():610] handle sender defer: 2
223
+ 2024-12-10 20:52:44,440 INFO SenderThread:597014 [sender.py:transition_state():614] send defer: 3
224
+ 2024-12-10 20:52:44,440 DEBUG SenderThread:597014 [sender.py:send():379] send: stats
225
+ 2024-12-10 20:52:44,440 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: defer
226
+ 2024-12-10 20:52:44,440 INFO HandlerThread:597014 [handler.py:handle_request_defer():184] handle defer: 3
227
+ 2024-12-10 20:52:44,440 DEBUG SenderThread:597014 [sender.py:send_request():406] send_request: defer
228
+ 2024-12-10 20:52:44,440 INFO SenderThread:597014 [sender.py:send_request_defer():610] handle sender defer: 3
229
+ 2024-12-10 20:52:44,440 INFO SenderThread:597014 [sender.py:transition_state():614] send defer: 4
230
+ 2024-12-10 20:52:44,441 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: defer
231
+ 2024-12-10 20:52:44,441 INFO HandlerThread:597014 [handler.py:handle_request_defer():184] handle defer: 4
232
+ 2024-12-10 20:52:44,670 INFO Thread-12 :597014 [dir_watcher.py:_on_file_created():271] file/dir created: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/wandb/run-20241210_205130-svysfc56/files/wandb-summary.json
233
+ 2024-12-10 20:52:44,671 INFO Thread-12 :597014 [dir_watcher.py:_on_file_created():271] file/dir created: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/wandb/run-20241210_205130-svysfc56/files/media/videos/eval/video_0_b6b0b3ba17b7513a5df9.mp4
234
+ 2024-12-10 20:52:44,671 INFO Thread-12 :597014 [dir_watcher.py:_on_file_created():271] file/dir created: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/wandb/run-20241210_205130-svysfc56/files/media/videos/eval
235
+ 2024-12-10 20:52:44,671 INFO Thread-12 :597014 [dir_watcher.py:_on_file_created():271] file/dir created: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/wandb/run-20241210_205130-svysfc56/files/media/videos
236
+ 2024-12-10 20:52:44,671 INFO Thread-12 :597014 [dir_watcher.py:_on_file_created():271] file/dir created: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/wandb/run-20241210_205130-svysfc56/files/media
237
+ 2024-12-10 20:52:44,696 INFO wandb-upload_1:597014 [upload_job.py:push():130] Uploaded file /tmp/tmp01kl0ofywandb/6w704qfx-media/videos/eval/video_0_b6b0b3ba17b7513a5df9.mp4
238
+ 2024-12-10 20:52:45,030 INFO wandb-upload_2:597014 [upload_job.py:push():130] Uploaded file /tmp/tmp01kl0ofywandb/kk0w0e5k-media/videos/eval/video_0_b6b0b3ba17b7513a5df9.mp4
239
+ 2024-12-10 20:52:45,671 INFO Thread-12 :597014 [dir_watcher.py:_on_file_modified():288] file/dir modified: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/wandb/run-20241210_205130-svysfc56/files/output.log
240
+ 2024-12-10 20:52:51,156 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: poll_exit
241
+ 2024-12-10 20:52:51,156 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: status_report
242
+ 2024-12-10 20:52:51,156 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: keepalive
243
+ 2024-12-10 20:52:51,157 DEBUG SenderThread:597014 [sender.py:send_request():406] send_request: defer
244
+ 2024-12-10 20:52:51,157 INFO SenderThread:597014 [sender.py:send_request_defer():610] handle sender defer: 4
245
+ 2024-12-10 20:52:51,157 INFO SenderThread:597014 [sender.py:transition_state():614] send defer: 5
246
+ 2024-12-10 20:52:51,157 DEBUG SenderThread:597014 [sender.py:send_request():406] send_request: poll_exit
247
+ 2024-12-10 20:52:51,157 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: defer
248
+ 2024-12-10 20:52:51,157 INFO HandlerThread:597014 [handler.py:handle_request_defer():184] handle defer: 5
249
+ 2024-12-10 20:52:51,157 DEBUG SenderThread:597014 [sender.py:send():379] send: summary
250
+ 2024-12-10 20:52:51,158 INFO SenderThread:597014 [sender.py:_save_file():1454] saving file wandb-summary.json with policy end
251
+ 2024-12-10 20:52:51,158 DEBUG SenderThread:597014 [sender.py:send_request():406] send_request: defer
252
+ 2024-12-10 20:52:51,158 INFO SenderThread:597014 [sender.py:send_request_defer():610] handle sender defer: 5
253
+ 2024-12-10 20:52:51,158 INFO SenderThread:597014 [sender.py:transition_state():614] send defer: 6
254
+ 2024-12-10 20:52:51,158 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: defer
255
+ 2024-12-10 20:52:51,158 INFO HandlerThread:597014 [handler.py:handle_request_defer():184] handle defer: 6
256
+ 2024-12-10 20:52:51,158 DEBUG SenderThread:597014 [sender.py:send_request():406] send_request: defer
257
+ 2024-12-10 20:52:51,158 INFO SenderThread:597014 [sender.py:send_request_defer():610] handle sender defer: 6
258
+ 2024-12-10 20:52:51,158 INFO SenderThread:597014 [sender.py:transition_state():614] send defer: 7
259
+ 2024-12-10 20:52:51,158 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: status_report
260
+ 2024-12-10 20:52:51,158 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: defer
261
+ 2024-12-10 20:52:51,158 INFO HandlerThread:597014 [handler.py:handle_request_defer():184] handle defer: 7
262
+ 2024-12-10 20:52:51,159 DEBUG SenderThread:597014 [sender.py:send_request():406] send_request: defer
263
+ 2024-12-10 20:52:51,159 INFO SenderThread:597014 [sender.py:send_request_defer():610] handle sender defer: 7
264
+ 2024-12-10 20:52:51,375 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: poll_exit
265
+ 2024-12-10 20:52:51,673 INFO Thread-12 :597014 [dir_watcher.py:_on_file_modified():288] file/dir modified: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/wandb/run-20241210_205130-svysfc56/files/wandb-summary.json
266
+ 2024-12-10 20:52:51,921 INFO SenderThread:597014 [sender.py:transition_state():614] send defer: 8
267
+ 2024-12-10 20:52:51,922 DEBUG SenderThread:597014 [sender.py:send_request():406] send_request: poll_exit
268
+ 2024-12-10 20:52:51,922 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: defer
269
+ 2024-12-10 20:52:51,922 INFO HandlerThread:597014 [handler.py:handle_request_defer():184] handle defer: 8
270
+ 2024-12-10 20:52:51,922 DEBUG SenderThread:597014 [sender.py:send_request():406] send_request: defer
271
+ 2024-12-10 20:52:51,922 INFO SenderThread:597014 [sender.py:send_request_defer():610] handle sender defer: 8
272
+ 2024-12-10 20:52:51,922 INFO SenderThread:597014 [job_builder.py:build():440] Attempting to build job artifact
273
+ 2024-12-10 20:52:51,924 INFO SenderThread:597014 [job_builder.py:_get_source_type():569] is repo sourced job
274
+ 2024-12-10 20:52:51,942 INFO SenderThread:597014 [job_builder.py:build():545] adding wandb-job metadata file
275
+ 2024-12-10 20:52:51,951 INFO SenderThread:597014 [sender.py:transition_state():614] send defer: 9
276
+ 2024-12-10 20:52:51,952 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: defer
277
+ 2024-12-10 20:52:51,952 INFO HandlerThread:597014 [handler.py:handle_request_defer():184] handle defer: 9
278
+ 2024-12-10 20:52:51,952 DEBUG SenderThread:597014 [sender.py:send():379] send: artifact
279
+ 2024-12-10 20:52:52,376 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: poll_exit
280
+ 2024-12-10 20:52:52,626 INFO wandb-upload_0:597014 [upload_job.py:push():85] Skipped uploading /home/leonmkim/.local/share/wandb/artifacts/staging/tmptoffctew
281
+ 2024-12-10 20:52:52,673 INFO Thread-12 :597014 [dir_watcher.py:_on_file_modified():288] file/dir modified: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/wandb/run-20241210_205130-svysfc56/files/output.log
282
+ 2024-12-10 20:52:52,858 INFO wandb-upload_2:597014 [upload_job.py:push():88] Uploaded file /tmp/tmpemx2d4yl/wandb-job.json
283
+ 2024-12-10 20:52:52,880 INFO wandb-upload_1:597014 [upload_job.py:push():88] Uploaded file /home/leonmkim/.local/share/wandb/artifacts/staging/tmpc2y7tffk
284
+ 2024-12-10 20:52:53,418 INFO SenderThread:597014 [sender.py:send_artifact():1537] sent artifact job-https___github.com_leonmkim_fish_leon.git_FISH_eval_robot.py - {'id': 'QXJ0aWZhY3Q6MTM2Njc1OTg1Ng==', 'state': 'PENDING', 'artifactSequence': {'id': 'QXJ0aWZhY3RDb2xsZWN0aW9uOjMxNzk5NjkyNQ==', 'latestArtifact': {'id': 'QXJ0aWZhY3Q6MTM1MTcyNjQ1OA==', 'versionIndex': 335}}}
285
+ 2024-12-10 20:52:53,418 DEBUG SenderThread:597014 [sender.py:send_request():406] send_request: defer
286
+ 2024-12-10 20:52:53,418 INFO SenderThread:597014 [sender.py:send_request_defer():610] handle sender defer: 9
287
+ 2024-12-10 20:52:53,418 INFO SenderThread:597014 [dir_watcher.py:finish():358] shutting down directory watcher
288
+ 2024-12-10 20:52:53,674 INFO SenderThread:597014 [dir_watcher.py:finish():388] scan: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/wandb/run-20241210_205130-svysfc56/files
289
+ 2024-12-10 20:52:53,674 INFO SenderThread:597014 [dir_watcher.py:finish():402] scan save: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/wandb/run-20241210_205130-svysfc56/files/config.yaml config.yaml
290
+ 2024-12-10 20:52:53,674 INFO SenderThread:597014 [dir_watcher.py:finish():402] scan save: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/wandb/run-20241210_205130-svysfc56/files/wandb-summary.json wandb-summary.json
291
+ 2024-12-10 20:52:53,674 INFO SenderThread:597014 [dir_watcher.py:finish():402] scan save: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/wandb/run-20241210_205130-svysfc56/files/requirements.txt requirements.txt
292
+ 2024-12-10 20:52:53,674 INFO SenderThread:597014 [dir_watcher.py:finish():402] scan save: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/wandb/run-20241210_205130-svysfc56/files/wandb-metadata.json wandb-metadata.json
293
+ 2024-12-10 20:52:53,674 INFO SenderThread:597014 [dir_watcher.py:finish():402] scan save: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/wandb/run-20241210_205130-svysfc56/files/events.out.tfevents.1733881891.leonmkim-ROG-Strix-G15CS-G15CS.596892.0 events.out.tfevents.1733881891.leonmkim-ROG-Strix-G15CS-G15CS.596892.0
294
+ 2024-12-10 20:52:53,674 INFO SenderThread:597014 [dir_watcher.py:finish():402] scan save: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/wandb/run-20241210_205130-svysfc56/files/output.log output.log
295
+ 2024-12-10 20:52:53,676 INFO SenderThread:597014 [dir_watcher.py:finish():402] scan save: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/wandb/run-20241210_205130-svysfc56/files/diff.patch diff.patch
296
+ 2024-12-10 20:52:53,680 INFO SenderThread:597014 [dir_watcher.py:finish():402] scan save: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/wandb/run-20241210_205130-svysfc56/files/code/FISH/eval_robot.py code/FISH/eval_robot.py
297
+ 2024-12-10 20:52:53,680 INFO SenderThread:597014 [dir_watcher.py:finish():402] scan save: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/wandb/run-20241210_205130-svysfc56/files/media/videos/eval/video_0_b6b0b3ba17b7513a5df9.mp4 media/videos/eval/video_0_b6b0b3ba17b7513a5df9.mp4
298
+ 2024-12-10 20:52:53,680 INFO SenderThread:597014 [sender.py:transition_state():614] send defer: 10
299
+ 2024-12-10 20:52:53,680 DEBUG SenderThread:597014 [sender.py:send_request():406] send_request: poll_exit
300
+ 2024-12-10 20:52:53,681 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: defer
301
+ 2024-12-10 20:52:53,681 INFO HandlerThread:597014 [handler.py:handle_request_defer():184] handle defer: 10
302
+ 2024-12-10 20:52:53,681 DEBUG SenderThread:597014 [sender.py:send_request():406] send_request: defer
303
+ 2024-12-10 20:52:53,681 INFO SenderThread:597014 [sender.py:send_request_defer():610] handle sender defer: 10
304
+ 2024-12-10 20:52:53,681 INFO SenderThread:597014 [file_pusher.py:finish():169] shutting down file pusher
305
+ 2024-12-10 20:52:53,872 INFO wandb-upload_2:597014 [upload_job.py:push():130] Uploaded file /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/wandb/run-20241210_205130-svysfc56/files/requirements.txt
306
+ 2024-12-10 20:52:53,888 INFO wandb-upload_0:597014 [upload_job.py:push():130] Uploaded file /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/wandb/run-20241210_205130-svysfc56/files/config.yaml
307
+ 2024-12-10 20:52:53,899 INFO wandb-upload_3:597014 [upload_job.py:push():130] Uploaded file /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/wandb/run-20241210_205130-svysfc56/files/wandb-summary.json
308
+ 2024-12-10 20:52:53,910 INFO wandb-upload_1:597014 [upload_job.py:push():130] Uploaded file /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/wandb/run-20241210_205130-svysfc56/files/output.log
309
+ 2024-12-10 20:52:54,110 INFO Thread-11 (_thread_body):597014 [sender.py:transition_state():614] send defer: 11
310
+ 2024-12-10 20:52:54,110 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: defer
311
+ 2024-12-10 20:52:54,111 INFO HandlerThread:597014 [handler.py:handle_request_defer():184] handle defer: 11
312
+ 2024-12-10 20:52:54,111 DEBUG SenderThread:597014 [sender.py:send_request():406] send_request: defer
313
+ 2024-12-10 20:52:54,111 INFO SenderThread:597014 [sender.py:send_request_defer():610] handle sender defer: 11
314
+ 2024-12-10 20:52:54,111 INFO SenderThread:597014 [file_pusher.py:join():175] waiting for file pusher
315
+ 2024-12-10 20:52:54,111 INFO SenderThread:597014 [sender.py:transition_state():614] send defer: 12
316
+ 2024-12-10 20:52:54,111 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: defer
317
+ 2024-12-10 20:52:54,111 INFO HandlerThread:597014 [handler.py:handle_request_defer():184] handle defer: 12
318
+ 2024-12-10 20:52:54,112 DEBUG SenderThread:597014 [sender.py:send_request():406] send_request: defer
319
+ 2024-12-10 20:52:54,112 INFO SenderThread:597014 [sender.py:send_request_defer():610] handle sender defer: 12
320
+ 2024-12-10 20:52:54,112 INFO SenderThread:597014 [file_stream.py:finish():601] file stream finish called
321
+ 2024-12-10 20:52:54,317 INFO SenderThread:597014 [file_stream.py:finish():605] file stream finish is done
322
+ 2024-12-10 20:52:54,317 INFO SenderThread:597014 [sender.py:transition_state():614] send defer: 13
323
+ 2024-12-10 20:52:54,318 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: defer
324
+ 2024-12-10 20:52:54,318 INFO HandlerThread:597014 [handler.py:handle_request_defer():184] handle defer: 13
325
+ 2024-12-10 20:52:54,318 DEBUG SenderThread:597014 [sender.py:send_request():406] send_request: defer
326
+ 2024-12-10 20:52:54,318 INFO SenderThread:597014 [sender.py:send_request_defer():610] handle sender defer: 13
327
+ 2024-12-10 20:52:54,318 INFO SenderThread:597014 [sender.py:transition_state():614] send defer: 14
328
+ 2024-12-10 20:52:54,318 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: defer
329
+ 2024-12-10 20:52:54,318 INFO HandlerThread:597014 [handler.py:handle_request_defer():184] handle defer: 14
330
+ 2024-12-10 20:52:54,319 DEBUG SenderThread:597014 [sender.py:send():379] send: final
331
+ 2024-12-10 20:52:54,319 DEBUG SenderThread:597014 [sender.py:send():379] send: footer
332
+ 2024-12-10 20:52:54,319 DEBUG SenderThread:597014 [sender.py:send_request():406] send_request: defer
333
+ 2024-12-10 20:52:54,319 INFO SenderThread:597014 [sender.py:send_request_defer():610] handle sender defer: 14
334
+ 2024-12-10 20:52:54,319 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: poll_exit
335
+ 2024-12-10 20:52:54,320 DEBUG SenderThread:597014 [sender.py:send_request():406] send_request: poll_exit
336
+ 2024-12-10 20:52:54,320 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: server_info
337
+ 2024-12-10 20:52:54,320 DEBUG SenderThread:597014 [sender.py:send_request():406] send_request: server_info
338
+ 2024-12-10 20:52:54,322 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: poll_exit
339
+ 2024-12-10 20:52:54,322 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: get_summary
340
+ 2024-12-10 20:52:54,322 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: sampled_history
341
+ 2024-12-10 20:52:54,322 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
342
+ 2024-12-10 20:52:54,378 DEBUG SenderThread:597014 [sender.py:send_request():406] send_request: poll_exit
343
+ 2024-12-10 20:52:54,378 INFO MainThread:597014 [wandb_run.py:_footer_history_summary_info():4016] rendering history
344
+ 2024-12-10 20:52:54,378 INFO MainThread:597014 [wandb_run.py:_footer_history_summary_info():4048] rendering summary
345
+ 2024-12-10 20:52:54,379 INFO MainThread:597014 [wandb_run.py:_footer_sync_info():3975] logging synced files
346
+ 2024-12-10 20:52:54,379 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: shutdown
347
+ 2024-12-10 20:52:54,379 INFO HandlerThread:597014 [handler.py:finish():882] shutting down handler
348
+ 2024-12-10 20:52:55,322 INFO WriterThread:597014 [datastore.py:close():296] close: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/wandb/run-20241210_205130-svysfc56/run-svysfc56.wandb
349
+ 2024-12-10 20:52:55,378 INFO SenderThread:597014 [sender.py:finish():1615] shutting down sender
350
+ 2024-12-10 20:52:55,378 INFO SenderThread:597014 [file_pusher.py:finish():169] shutting down file pusher
351
+ 2024-12-10 20:52:55,378 INFO SenderThread:597014 [file_pusher.py:join():175] waiting for file pusher
205123/wandb/debug.log ADDED
@@ -0,0 +1,31 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ 2024-12-10 20:51:30,303 INFO MainThread:596892 [wandb_setup.py:_flush():76] Current SDK version is 0.17.5
2
+ 2024-12-10 20:51:30,303 INFO MainThread:596892 [wandb_setup.py:_flush():76] Configure stats pid to 596892
3
+ 2024-12-10 20:51:30,303 INFO MainThread:596892 [wandb_setup.py:_flush():76] Loading settings from /home/leonmkim/.config/wandb/settings
4
+ 2024-12-10 20:51:30,303 INFO MainThread:596892 [wandb_setup.py:_flush():76] Loading settings from /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/wandb/settings
5
+ 2024-12-10 20:51:30,303 INFO MainThread:596892 [wandb_setup.py:_flush():76] Loading settings from environment variables: {}
6
+ 2024-12-10 20:51:30,303 INFO MainThread:596892 [wandb_setup.py:_flush():76] Applying setup settings: {'_disable_service': False}
7
+ 2024-12-10 20:51:30,304 INFO MainThread:596892 [wandb_setup.py:_flush():76] Inferring run settings from compute environment: {'program_relpath': 'FISH/eval_robot.py', 'program_abspath': '/home/leonmkim/fish_leon/FISH/eval_robot.py', 'program': '/home/leonmkim/fish_leon/FISH/eval_robot.py'}
8
+ 2024-12-10 20:51:30,304 INFO MainThread:596892 [wandb_setup.py:_flush():76] Applying login settings: {}
9
+ 2024-12-10 20:51:30,304 INFO MainThread:596892 [wandb_init.py:_log_setup():529] Logging user logs to /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/wandb/run-20241210_205130-svysfc56/logs/debug.log
10
+ 2024-12-10 20:51:30,304 INFO MainThread:596892 [wandb_init.py:_log_setup():530] Logging internal logs to /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/wandb/run-20241210_205130-svysfc56/logs/debug-internal.log
11
+ 2024-12-10 20:51:30,304 INFO MainThread:596892 [wandb_init.py:init():569] calling init triggers
12
+ 2024-12-10 20:51:30,304 INFO MainThread:596892 [wandb_init.py:init():576] wandb.init called with sweep_config: {}
13
+ config: {'root_dir': '/home/leonmkim/fish_leon', 'replay_buffer_size': 150000, 'replay_buffer_num_workers': 2, 'nstep': 3, 'batch_size': 128, 'seed': 0, 'dataset_shuffle_seed': 0, 'device': 'cuda', 'save_video': True, 'save_train_video': True, 'use_tb': True, 'use_wandb': True, 'wandb_run_id': '265626', 'wandb_notes': '265626_req_265827', 'eval': True, 'process_contact_features': True, 'obs_type': 'pixels', 'use_color': True, 'use_depth': True, 'use_masks': True, 'mask_list': ['EE_obj_mask'], 'mask_representation': 'channels', 'crop_hw': [144, 144], 'crop_down_offset': 48, 'color_crop_type': None, 'depth_crop_type': None, 'segmask_crop_type': None, 'add_crop_binary_mask': False, 'add_coord_conv_map': False, 'use_context_color': False, 'use_context_depth': False, 'use_context_segmask': False, 'context_color_crop_type': None, 'context_depth_crop_type': None, 'context_segmask_crop_type': None, 'context_add_crop_binary_mask': False, 'context_add_coord_conv_map': False, 'use_contact_map': False, 'use_sdf_maps': False, 'use_normals_maps': False, 'which_objects': 'both', 'max_contact_prob': 0.1, 'max_depth': 2.0, 'grasped_dtc_max_value': 0.2, 'env_dtc_max_value': 0.4, 'grasped_normals_mask_max_dtc_value': 0.2, 'env_normals_mask_max_dtc_value': 0.4, 'clamp_dtc': True, 'dtc_adaptive_normalization': False, 'mask_normals_within_sdf': True, 'adaptive_normals_mask': True, 'learnable_contact_preprocess_params': True, 'contact_model_name': 'local_multitask_outhd64all_home_crop_h144w144d48_mask_ctxtmask_seed_220979_epoch_9', 'contact_estimation_model_ckpt_path': '~/fish_leon/contact_estimation/artifacts/175604_2/checkpoints/epoch=09-val_loss=0.00.ckpt', 'encoder_type': 'small', 'debug_timestamps': False, 'open_loop': False, 'action_trajectories': True, 'stop_after_action': False, 'interpolation_frequency': 25, 'policy_frequency': 5, 'wait_for_new_camera_frames': True, 'baseline': False, 'train_demo_idxs_list_or_num': -1, 'log_train_every_steps': 25, 'name_of_expert_demo': '64_240x320_hbm_twodim_fps_fix_contact_fix_recovery_fix_annotated_start_idx_5hz_zstd7_EE_pxl_coords_expert_demos_imp_act', 'expert_dataset_dirpath': '/home/leonmkim/fish_leon/FISH/expert_demos/frankagym/FrankaInsertion-v1/64_240x320_hbm_twodim_fps_fix_contact_fix_recovery_fix_annotated_start_idx_5hz_zstd7_EE_pxl_coords_expert_demos_imp_act', 'store_dataset_in_memory': False, 'expert_dataset': '/home/leonmkim/fish_leon/FISH/expert_demos/frankagym/FrankaInsertion-v1/64_240x320_hbm_twodim_fps_fix_contact_fix_recovery_fix_annotated_start_idx_5hz_zstd7_EE_pxl_coords_expert_demos_imp_act/demos.zarr', 'action_key': 'action_trajectory_25hz', 'semantic_demo_grouping_name': 'semantic_demo_grouping.yaml', 'semantic_demo_grouping': '/home/leonmkim/fish_leon/FISH/expert_demos/frankagym/FrankaInsertion-v1/64_240x320_hbm_twodim_fps_fix_contact_fix_recovery_fix_annotated_start_idx_5hz_zstd7_EE_pxl_coords_expert_demos_imp_act/semantic_demo_grouping.yaml', 'include_groups_list': ['hbm_twodim_left_to_right'], 'expert_dataset_config': '/home/leonmkim/fish_leon/FISH/expert_demos/frankagym/FrankaInsertion-v1/64_240x320_hbm_twodim_fps_fix_contact_fix_recovery_fix_annotated_start_idx_5hz_zstd7_EE_pxl_coords_expert_demos_imp_act/demo_config.yaml', 'name_of_valid_demo': '64_240x320_hbm_twodim_fps_fix_contact_fix_recovery_fix_annotated_start_idx_5hz_zstd7_EE_pxl_coords_expert_demos_imp_act', 'valid_dataset_dir': '/home/leonmkim/fish_leon/FISH/expert_demos/frankagym/FrankaInsertion-v1/64_240x320_hbm_twodim_fps_fix_contact_fix_recovery_fix_annotated_start_idx_5hz_zstd7_EE_pxl_coords_expert_demos_imp_act/demos.zarr', 'valid_demo_idxs_list_or_num': None, 'val_num_groups': 0, 'load_bc': True, 'checkpoint_epoch_list': [99, 199, 299, 399, 499, 599, 699, 799, 899, 999, 1249, 1499, 1749, 1999, 2999, 3999, 4999, 5999, 6999, 7999, 8999, 9999], 'snapshot_root_dir': '/mnt/grasp_high_usage/leonmkim/contact_estimation/FISH', 'save_snapshot': True, 'save_last_snapshot': True, 'save_snapshot_when_done': True, 'top_k_checkpoints': 5, 'save_snapshot_link_to_weights_dir': 'deprecated', 'bc_regularize': False, 'bc_weight_type': 'qfilter', 'experiment_dir': './exp_local/frankagym_pixels/FrankaInsertion-v1/265626', 'agent': {'_target_': 'agent.diffusion_policy.DiffusionPolicyAgent', 'name': 'diffusion_policy', 'load_checkpoint': True, 'obs_shape': [5, 240, 320], 'action_shape': [36, 7], 'device': 'cuda', 'n_obs_steps': 1, 'suite_name': 'frankagym', 'obs_type': 'pixels', 'enable_arm': True, 'enable_camera': True, 'use_tb': True, 'desired_image_shape': [13, 180, 240], 'orig_cam_shape': [3, 240, 320], 'config': {'_target_': 'agent.diffusion_policy.DiffusionPolicyAgentConfig', 'compile': False, 'device': 'cuda', 'cam_resize_shape': [13, 180, 240], 'orig_cam_shape': [3, 240, 320], 'policy_cfg': {'_target_': 'lerobot.common.policies.diffusion.configuration_diffusion.DiffusionConfig', 'n_obs_steps': 1, 'horizon': 36, 'n_action_steps': 36, 'output_shapes': {'action': [7]}, 'input_normalization_modes': {'observation.image': 'mean_std', 'observation.state': 'min_max', 'observation.action_history': 'min_max'}, 'output_normalization_modes': {'action': 'min_max'}, 'vision_backbone': 'resnet18', 'pretrained_backbone_weights': None, 'transforms': [{'_target_': 'torchaug.transforms.RandomAffine', 'degrees': [-5, 5], 'translate': [0.05, 0.05], 'batch_transform': True, 'num_chunks': -1, 'batch_inplace': True}, {'_target_': 'torchaug.transforms.RandomColorJitter', 'brightness': 0.3, 'contrast': 0.4, 'saturation': 0.5, 'hue': 0.08, 'batch_transform': True, 'num_chunks': -1, 'batch_inplace': True}], 'use_group_norm': True, 'spatial_softmax_num_keypoints': 32, 'action_history_encoder_config': {'_target_': 'lerobot.common.policies.diffusion.configuration_diffusion.Unet1dEncoderConfig', 'in_channels': 7, 'out_channels': 32, 'history_length': 6, 'kernel_size': 5, 'downsample_kernel_size': 3, 'downsample_stride': 2, 'downsample_padding': 1}, 'down_dims': [256, 512, 1024], 'kernel_size': 5, 'n_groups': 8, 'diffusion_step_embed_dim': 128, 'use_film_scale_modulation': True, 'noise_scheduler_type': 'DDIM', 'beta_schedule': 'squaredcos_cap_v2', 'beta_start': 0.0001, 'beta_end': 0.02, 'prediction_type': 'epsilon', 'clip_sample': True, 'clip_sample_range': 1.0, 'num_train_timesteps': 50, 'num_inference_steps': 10, 'do_mask_loss_for_padding': False, 'input_shapes': {'observation.image': [13, 180, 240], 'context_observation.image': [13, 180, 240], 'observation.state': [8], 'observation.action_history': [7]}}, 'train_cfg': {'_target_': 'utils.TrainConfig', 'lr': 0.0001, 'lr_scheduler': 'cosine', 'lr_warmup_steps': 500, 'adam_betas': [0.95, 0.999], 'adam_eps': 1e-08, 'adam_weight_decay': 1e-06, 'grad_clip_norm': 10, 'offline_steps': 1000000, 'use_amp': True}, 'observation_cfg': {'_target_': 'agent.encoder.VisualFeatureSet', 'use_depth': True, 'use_color': True, 'mask_input_dict': {'_target_': 'agent.encoder.MaskInputDict', 'enable': True, 'representation': 'channels', 'mask_list': ['EE_obj_mask']}, 'crop_input_config': {'_target_': 'agent.encoder.CropInputConfig', 'color_crop_type': None, 'depth_crop_type': None, 'segmask_crop_type': None, 'crop_hw': [144, 144], 'crop_down_offset': 48, 'add_crop_binary_mask': False, 'add_coord_conv_map': False}, 'context_input_config': {'_target_': 'agent.encoder.ContextInputConfig', 'use_color': False, 'use_depth': False, 'mask_input_dict': {'_target_': 'agent.encoder.MaskInputDict', 'enable': False, 'representation': 'channels', 'mask_list': ['EE_obj_mask']}, 'crop_input_config': {'_target_': 'agent.encoder.CropInputConfig', 'color_crop_type': None, 'depth_crop_type': None, 'segmask_crop_type': None, 'crop_hw': [144, 144], 'crop_down_offset': 48, 'add_crop_binary_mask': False, 'add_coord_conv_map': False}}, 'mask_soft_approx_scheduler_config': {'_target_': 'agent.encoder.MaskSoftApproxSchedulerConfig', 'num_steps': 40000, 'initial_value': 10.0, 'final_value': 1000.0, 'interpolation_scheme': 'cosine'}, 'use_contact_map': False, 'use_sdf_maps': False, 'use_normals_maps': False, 'which_objects': 'both', 'grasped_dtc_max_value': 0.2, 'env_dtc_max_value': 0.4, 'grasped_normals_mask_max_dtc_value': 0.2, 'env_normals_mask_max_dtc_value': 0.4, 'clamp_dtc': True, 'max_contact_prob': 0.1, 'mask_normals_within_sdf': True, 'dtc_adaptive_normalization': False, 'adaptive_normals_mask': True, 'max_depth': 2.0, 'image_shape': [13, 180, 240], 'learnable_contact_preprocess_params': True, 'learning_rate': 0.0001, 'weight_decay': 0.0, 'contact_model_name': 'local_multitask_outhd64all_home_crop_h144w144d48_mask_ctxtmask_seed_220979_epoch_9', 'zero_centered': False}, 'policy_frequency': 5, 'interpolation_frequency': 25}}, 'suite': {'suite': 'frankagym', 'name': 'frankagym', 'frame_stack': 1, 'action_repeat': 1, 'discount': 0.99, 'hidden_dim': 1024, 'num_train_frames': 2010, 'num_seed_frames': 260, 'num_train_epochs': 5000, 'validate_every_epochs': 100, 'validate_diffusion_on_action_loss_every_epochs': 500, 'train_eval_diffusion_on_action_loss_every_epochs': 500, 'check_topk_every_epochs': 10, 'save_snapshot_every_epochs': 5000, 'eval_every_frames': 2000, 'num_eval_episodes': 5, 'save_snapshot': True, 'wait_for_user_to_start_episode': True, 'task_make_fn': {'_target_': 'suite.frankagym.make', 'name': 'FrankaInsertion-v1', 'height': 240, 'width': 320, 'frame_stack': 1, 'action_repeat': 1, 'seed': 0, 'enable_arm': True, 'enable_gripper': True, 'start_with_gripper_open': True, 'enable_camera': True, 'path_to_depth_extrinsics': '~/fish_leon/FISH/cfgs/camera_poses/camera_poses_L515/20240904-122305/color_tf_world.npy', 'contact_estimation_model_ckpt_path': '~/fish_leon/contact_estimation/artifacts/175604_2/checkpoints/epoch=09-val_loss=0.00.ckpt', 'x_limit': [0.2, 0.7], 'y_limit': [-0.4, 0.4], 'z_limit': [-0.05, 0.55], 'device': 'cuda', 'interpolation_frequency': 25, 'policy_frequency': 5, 'debug_timestamps': False, 'stop_after_action': False, 'open_loop': False, 'wait_for_new_camera_frames': True, 'action_key': 'action_trajectory_25hz', 'action_trajectory_horizon': 36, 'action_trajectories': True, 'path_to_zarr_dataset': '/home/leonmkim/fish_leon/FISH/expert_demos/frankagym/FrankaInsertion-v1/64_240x320_hbm_twodim_fps_fix_contact_fix_recovery_fix_annotated_start_idx_5hz_zstd7_EE_pxl_coords_expert_demos_imp_act/demos.zarr', 'observation_cfg': {'_target_': 'agent.encoder.VisualFeatureSet', 'use_depth': True, 'use_color': True, 'mask_input_dict': {'_target_': 'agent.encoder.MaskInputDict', 'enable': True, 'representation': 'channels', 'mask_list': ['EE_obj_mask']}, 'crop_input_config': {'_target_': 'agent.encoder.CropInputConfig', 'color_crop_type': None, 'depth_crop_type': None, 'segmask_crop_type': None, 'crop_hw': [144, 144], 'crop_down_offset': 48, 'add_crop_binary_mask': False, 'add_coord_conv_map': False}, 'context_input_config': {'_target_': 'agent.encoder.ContextInputConfig', 'use_color': False, 'use_depth': False, 'mask_input_dict': {'_target_': 'agent.encoder.MaskInputDict', 'enable': False, 'representation': 'channels', 'mask_list': ['EE_obj_mask']}, 'crop_input_config': {'_target_': 'agent.encoder.CropInputConfig', 'color_crop_type': None, 'depth_crop_type': None, 'segmask_crop_type': None, 'crop_hw': [144, 144], 'crop_down_offset': 48, 'add_crop_binary_mask': False, 'add_coord_conv_map': False}}, 'mask_soft_approx_scheduler_config': {'_target_': 'agent.encoder.MaskSoftApproxSchedulerConfig', 'num_steps': 40000, 'initial_value': 10.0, 'final_value': 1000.0, 'interpolation_scheme': 'cosine'}, 'use_contact_map': False, 'use_sdf_maps': False, 'use_normals_maps': False, 'which_objects': 'both', 'grasped_dtc_max_value': 0.2, 'env_dtc_max_value': 0.4, 'grasped_normals_mask_max_dtc_value': 0.2, 'env_normals_mask_max_dtc_value': 0.4, 'clamp_dtc': True, 'max_contact_prob': 0.1, 'mask_normals_within_sdf': True, 'dtc_adaptive_normalization': False, 'adaptive_normals_mask': True, 'max_depth': 2.0, 'image_shape': [13, 180, 240], 'learnable_contact_preprocess_params': True, 'learning_rate': 0.0001, 'weight_decay': 0.0, 'contact_model_name': 'local_multitask_outhd64all_home_crop_h144w144d48_mask_ctxtmask_seed_220979_epoch_9', 'zero_centered': False}, 'agent_policy_cfg': {'_target_': 'agent.diffusion_policy.DiffusionPolicyAgentConfig', 'compile': False, 'device': 'cuda', 'cam_resize_shape': [13, 180, 240], 'orig_cam_shape': [3, 240, 320], 'policy_cfg': {'_target_': 'lerobot.common.policies.diffusion.configuration_diffusion.DiffusionConfig', 'n_obs_steps': 1, 'horizon': 36, 'n_action_steps': 36, 'output_shapes': {'action': [7]}, 'input_normalization_modes': {'observation.image': 'mean_std', 'observation.state': 'min_max', 'observation.action_history': 'min_max'}, 'output_normalization_modes': {'action': 'min_max'}, 'vision_backbone': 'resnet18', 'pretrained_backbone_weights': None, 'transforms': [{'_target_': 'torchaug.transforms.RandomAffine', 'degrees': [-5, 5], 'translate': [0.05, 0.05], 'batch_transform': True, 'num_chunks': -1, 'batch_inplace': True}, {'_target_': 'torchaug.transforms.RandomColorJitter', 'brightness': 0.3, 'contrast': 0.4, 'saturation': 0.5, 'hue': 0.08, 'batch_transform': True, 'num_chunks': -1, 'batch_inplace': True}], 'use_group_norm': True, 'spatial_softmax_num_keypoints': 32, 'action_history_encoder_config': {'_target_': 'lerobot.common.policies.diffusion.configuration_diffusion.Unet1dEncoderConfig', 'in_channels': 7, 'out_channels': 32, 'history_length': 6, 'kernel_size': 5, 'downsample_kernel_size': 3, 'downsample_stride': 2, 'downsample_padding': 1}, 'down_dims': [256, 512, 1024], 'kernel_size': 5, 'n_groups': 8, 'diffusion_step_embed_dim': 128, 'use_film_scale_modulation': True, 'noise_scheduler_type': 'DDIM', 'beta_schedule': 'squaredcos_cap_v2', 'beta_start': 0.0001, 'beta_end': 0.02, 'prediction_type': 'epsilon', 'clip_sample': True, 'clip_sample_range': 1.0, 'num_train_timesteps': 50, 'num_inference_steps': 10, 'do_mask_loss_for_padding': False, 'input_shapes': {'observation.image': [13, 180, 240], 'context_observation.image': [13, 180, 240], 'observation.state': [8], 'observation.action_history': [7]}}, 'train_cfg': {'_target_': 'utils.TrainConfig', 'lr': 0.0001, 'lr_scheduler': 'cosine', 'lr_warmup_steps': 500, 'adam_betas': [0.95, 0.999], 'adam_eps': 1e-08, 'adam_weight_decay': 1e-06, 'grad_clip_norm': 10, 'offline_steps': 1000000, 'use_amp': True}, 'observation_cfg': {'_target_': 'agent.encoder.VisualFeatureSet', 'use_depth': True, 'use_color': True, 'mask_input_dict': {'_target_': 'agent.encoder.MaskInputDict', 'enable': True, 'representation': 'channels', 'mask_list': ['EE_obj_mask']}, 'crop_input_config': {'_target_': 'agent.encoder.CropInputConfig', 'color_crop_type': None, 'depth_crop_type': None, 'segmask_crop_type': None, 'crop_hw': [144, 144], 'crop_down_offset': 48, 'add_crop_binary_mask': False, 'add_coord_conv_map': False}, 'context_input_config': {'_target_': 'agent.encoder.ContextInputConfig', 'use_color': False, 'use_depth': False, 'mask_input_dict': {'_target_': 'agent.encoder.MaskInputDict', 'enable': False, 'representation': 'channels', 'mask_list': ['EE_obj_mask']}, 'crop_input_config': {'_target_': 'agent.encoder.CropInputConfig', 'color_crop_type': None, 'depth_crop_type': None, 'segmask_crop_type': None, 'crop_hw': [144, 144], 'crop_down_offset': 48, 'add_crop_binary_mask': False, 'add_coord_conv_map': False}}, 'mask_soft_approx_scheduler_config': {'_target_': 'agent.encoder.MaskSoftApproxSchedulerConfig', 'num_steps': 40000, 'initial_value': 10.0, 'final_value': 1000.0, 'interpolation_scheme': 'cosine'}, 'use_contact_map': False, 'use_sdf_maps': False, 'use_normals_maps': False, 'which_objects': 'both', 'grasped_dtc_max_value': 0.2, 'env_dtc_max_value': 0.4, 'grasped_normals_mask_max_dtc_value': 0.2, 'env_normals_mask_max_dtc_value': 0.4, 'clamp_dtc': True, 'max_contact_prob': 0.1, 'mask_normals_within_sdf': True, 'dtc_adaptive_normalization': False, 'adaptive_normals_mask': True, 'max_depth': 2.0, 'image_shape': [13, 180, 240], 'learnable_contact_preprocess_params': True, 'learning_rate': 0.0001, 'weight_decay': 0.0, 'contact_model_name': 'local_multitask_outhd64all_home_crop_h144w144d48_mask_ctxtmask_seed_220979_epoch_9', 'zero_centered': False}, 'policy_frequency': 5, 'interpolation_frequency': 25}, 'true_action_history': True}}, 'num_train_frames_bc': 50000, 'num_train_frames_drq': 1100000, 'stddev_schedule_drq': 'linear(1.0,0.1,100000)', 'task_name': 'FrankaInsertion-v1', 'num_train_frames_vinn': 25000, 'num_train_frames_diffusion': 1000000, 'num_train_epochs_bc': 5000, 'num_train_epochs_diffusion': 15000, 'validate_every_epochs_bc': 5, 'validate_every_epochs_diffusion': 250, 'validate_diffusion_on_action_loss_every_epochs': 250, 'train_eval_diffusion_on_action_loss_every_epochs': 250, 'check_topk_every_epochs': 5, 'check_topk_every_epochs_diffusion': 250, 'save_snapshot_every_epochs_diffusion': 1500, 'x_limit': [0.2, 0.7], 'y_limit': [-0.4, 0.4], 'z_limit': [-0.05, 0.55], 'home_displacement': [0.55, 0.0, 0.55, 180.0, 0.0, 0.0], 'enable_gripper': True, 'start_with_gripper_open': True, 'offset_mask': [1, 1, 1, 1, 1, 1], 'path_to_depth_extrinsics': '~/fish_leon/FISH/cfgs/camera_poses/camera_poses_L515/20240904-122305/color_tf_world.npy', 'feature_type': '180x240_1_RGB_D_2.0_msk_channels_EE_obj_mask_acthst_hst6_out32_dwnkrnl3_dwnstrd2_dwnpd1', 'save_buffer': True, 'num_eval': 5, 'random_start': False, 'eval_starts': '/home/leonmkim/fish_leon/FISH/eval_starts/frankagym_pixels/FrankaInsertion-v1', 'num_valid_demos': None, 'load_checkpoint': True, 'true_action_history': True, 'checkpoint_epoch': 12000, 'load_residual_weight': False, 'checkpoint_root_dir': '/home/leonmkim/fish_leon/FISH', 'checkpoint_weight_dir': '/home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626', 'residual_weight': '/home/leonmkim/fish_leon/FISH/weights/frankagym_pixels/FrankaInsertion-v1/weight.pt'}
14
+ 2024-12-10 20:51:30,304 INFO MainThread:596892 [wandb_init.py:init():619] starting backend
15
+ 2024-12-10 20:51:30,304 INFO MainThread:596892 [wandb_init.py:init():623] setting up manager
16
+ 2024-12-10 20:51:30,310 INFO MainThread:596892 [backend.py:_multiprocessing_setup():105] multiprocessing start_methods=fork,spawn,forkserver, using: spawn
17
+ 2024-12-10 20:51:30,312 INFO MainThread:596892 [wandb_init.py:init():631] backend started and connected
18
+ 2024-12-10 20:51:30,324 INFO MainThread:596892 [wandb_init.py:init():720] updated telemetry
19
+ 2024-12-10 20:51:30,334 INFO MainThread:596892 [wandb_init.py:init():753] communicating run to backend with 90.0 second timeout
20
+ 2024-12-10 20:51:30,656 INFO MainThread:596892 [wandb_run.py:_on_init():2435] communicating current version
21
+ 2024-12-10 20:51:30,737 INFO MainThread:596892 [wandb_run.py:_on_init():2444] got version response upgrade_message: "wandb version 0.19.0 is available! To upgrade, please run:\n $ pip install wandb --upgrade"
22
+
23
+ 2024-12-10 20:51:30,737 INFO MainThread:596892 [wandb_init.py:init():804] starting run threads in backend
24
+ 2024-12-10 20:51:31,144 INFO MainThread:596892 [wandb_run.py:_console_start():2413] atexit reg
25
+ 2024-12-10 20:51:31,145 INFO MainThread:596892 [wandb_run.py:_redirect():2255] redirect: wrap_raw
26
+ 2024-12-10 20:51:31,145 INFO MainThread:596892 [wandb_run.py:_redirect():2320] Wrapping output streams.
27
+ 2024-12-10 20:51:31,145 INFO MainThread:596892 [wandb_run.py:_redirect():2345] Redirects installed.
28
+ 2024-12-10 20:51:31,147 INFO MainThread:596892 [wandb_init.py:init():847] run started, returning control to user process
29
+ 2024-12-10 20:51:31,147 INFO MainThread:596892 [wandb_run.py:_tensorboard_callback():1544] tensorboard callback: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/tb, True
30
+ 2024-12-10 20:51:46,908 INFO MainThread:596892 [wandb_run.py:_config_callback():1382] config_cb None None {'grasped_obj_name': 'hbm', 'left_book_slot': 'twodim'}
31
+ 2024-12-10 20:52:55,379 WARNING MsgRouterThr:596892 [router.py:message_loop():77] message_loop has been closed
205123/wandb/run-20241210_205130-svysfc56/files/code/FISH/eval_robot.py ADDED
@@ -0,0 +1,512 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ #%%
2
+ import warnings
3
+ import os
4
+
5
+ os.environ['MKL_SERVICE_FORCE_INTEL'] = '1'
6
+ os.environ['MUJOCO_GL'] = 'egl'
7
+ from pathlib import Path
8
+ #%%
9
+ import hydra
10
+ import numpy as np
11
+ import torch
12
+
13
+ import utils
14
+ from utils import get_feature_dirname_from_configs
15
+
16
+ from video import VideoRecorder
17
+ import pickle
18
+ import time
19
+ import threading
20
+ import shutil
21
+ from logger import Logger
22
+
23
+ import wandb
24
+ from omegaconf import OmegaConf, open_dict
25
+
26
+ from replay_buffer_robot import RosbagEvalReplayBufferStorage
27
+ from lerobot.common.utils.utils import _relative_path_between
28
+
29
+ torch.backends.cudnn.benchmark = True
30
+ warnings.filterwarnings('ignore', category=DeprecationWarning)
31
+
32
+ # import specs for replay buffer
33
+ from dm_env import specs
34
+
35
+ import sys, signal
36
+ import yaml
37
+
38
+ # get path of current file
39
+ current_path = os.path.dirname(os.path.realpath(__file__))
40
+ sys.path.append(os.path.join(current_path, os.pardir))
41
+ # from contact_estimation.src.utils.viz_utils import normalized_surface_normal_to_rgb, depth_map_to_im, grasped_env_dtc_map_to_im, contact_prob_map_to_im, desaturate_color_image, masked_overlay_im_list
42
+
43
+ def make_agent(obs_spec, action_spec, cfg):
44
+ cfg.obs_shape = obs_spec['pixels'].shape
45
+ dataset_statistics = None # this will be loaded from the checkpoint
46
+ try:
47
+ cfg.action_shape = action_spec.shape
48
+ except:
49
+ pass
50
+ return hydra.utils.instantiate(cfg, dataset_statistics)
51
+
52
+ class Workspace:
53
+ def __init__(self, cfg):
54
+ self.work_dir = Path.cwd()
55
+ print(f'workspace: {self.work_dir}')
56
+
57
+ signal.signal(signal.SIGINT, self.signal_handler)
58
+
59
+ self.cfg = cfg
60
+ self.loading_uncompiled_checkpoint_with_compile = False
61
+ self.loading_compiled_checkpoint_with_no_compile = False
62
+
63
+ snapshot_path = Path(self.cfg.checkpoint_weight_dir) / f'snapshot_{self.cfg.checkpoint_epoch}.pt'
64
+ self.load_checkpoint_conf(snapshot_path=snapshot_path)
65
+
66
+ # load config for action trajectories
67
+ utils.set_seed_everywhere(self.cfg.seed)
68
+ self.device = torch.device(self.cfg.device)
69
+ self.setup()
70
+
71
+ # self.agent = make_agent(self.eval_env.observation_spec(),
72
+ # self.eval_env.action_spec(), self.cfg.agent)
73
+ self.timer = utils.Timer()
74
+ # self._global_step = 0
75
+ self._global_episode = 0
76
+ self._global_epoch = 0
77
+ self.num_episode_successes = 0
78
+
79
+ # Need to convert hydra config to primitive container for wandb https://docs.wandb.ai/guides/integrations/hydra
80
+ with open_dict(self.cfg):
81
+ self.cfg.feature_type = get_feature_dirname_from_configs(
82
+ hydra.utils.instantiate(self.cfg.agent.config.observation_cfg),
83
+ self.cfg.agent.config.policy_cfg.input_shapes,
84
+ hydra.utils.instantiate(self.cfg.agent.config.policy_cfg.action_history_encoder_config) if 'observation.action_history' in self.cfg.agent.config.policy_cfg.input_shapes else None,
85
+ )
86
+
87
+ wandb_config = OmegaConf.to_container(
88
+ self.cfg, resolve=True, throw_on_missing=True
89
+ )
90
+ # must be called before any tf summary writer is created
91
+ if self.cfg.use_wandb:
92
+ wandb.init(project='extrinsic_contact_downstream', entity='serialexperimentsleon', job_type='eval', sync_tensorboard=self.cfg.use_tb, config=wandb_config)
93
+
94
+ self.logger = Logger(self.work_dir, use_tb=self.cfg.use_tb, use_wandb=self.cfg.use_wandb)
95
+
96
+ # if not self.loading_uncompiled_checkpoint_with_compile and self.cfg.agent.config.compile:
97
+ # self.agent.compile_modules()
98
+
99
+ # self.load_checkpoint(snapshot_path=snapshot_path)
100
+
101
+ # if self.loading_uncompiled_checkpoint_with_compile: # need to call compile after loading the checkpoint
102
+ # self.agent.compile_modules()
103
+
104
+ print(f"loaded agent with feature_type: {self.cfg.feature_type}")
105
+
106
+ def check_for_key_press(self):
107
+ while self.continue_keypress_thread:
108
+ inp = input("Press 'r' to restart current episode, 'n' to stop current episode and skip to next, 'q' to break entire eval\n")
109
+ if inp == 'n':
110
+ self.preempt_episode = True
111
+ print("preempting episode")
112
+ elif inp in ['', '0', '1']: # enter key
113
+ if inp in ['0', '1']:
114
+ self.num_episode_successes += int(inp)
115
+ self.proceed_after_env_reset_event.set()
116
+ print("proceeding to start episode!")
117
+ elif inp == 'q':
118
+ self.proceed_after_env_reset_event.set()
119
+ self.preempt_episode = True
120
+ self.exit_eval = True
121
+ self.continue_keypress_thread = False # will stop the keypress thread
122
+ print("quitting eval")
123
+ break
124
+ elif inp == 'r':
125
+ print('restarting episode')
126
+ self.preempt_episode = True
127
+ self.restart_episode = True
128
+ else:
129
+ print("Invalid key press, try again")
130
+
131
+ # self.keypress_input_thread.join() # wait for the keypress thread to finish
132
+
133
+ def signal_handler(self, signal, frame):
134
+ print("\nprogram exiting gracefully")
135
+ self.proceed_after_env_reset_event.set()
136
+ self.preempt_episode = True
137
+ self.exit_eval = True
138
+ self.continue_keypress_thread = False # will stop the keypress thread
139
+ self.keypress_input_thread.join() # wait for the keypress thread to finish
140
+ video_filepath = self.video_recorder.save()
141
+ # get the video file and convert to video tensor to log
142
+ self.logger.log_video('eval/video', video_filepath, self.global_step)
143
+ sys.exit(0)
144
+
145
+ def setup(self):
146
+ # create envs
147
+ self.eval_env = hydra.utils.call(self.cfg.suite.task_make_fn)
148
+ # expert_demo_config_path = os.path.join(os.path.dirname(self.cfg.expert_dataset), 'demo_config.yaml')
149
+ # self.expert_demo_config = yaml.load(open(expert_demo_config_path, 'r'), Loader=yaml.FullLoader)
150
+ # self.eval_env._env.action_trans_norm = expert_demo_config['max_translation_action_norm']
151
+ # self.eval_env._env.action_rot_norm = expert_demo_config['max_rotation_action_norm']
152
+ # self.eval_env._env.action_period = expert_demo_config['sample_period']
153
+ # print(f"setting max_translation_action_norm to {expert_demo_config['max_translation_action_norm']} and sample_period to {expert_demo_config['sample_period']}")
154
+ # print(f"setting max_rotation_action_norm to {expert_demo_config['max_rotation_action_norm']}")
155
+
156
+ # self.eval_env.set_demo_params(self.cfg.expert_dataset)
157
+
158
+ # Turn off random start
159
+ self.eval_env.random_start = False
160
+
161
+ # create replay buffer
162
+ # data_specs = [
163
+ # {
164
+ # 'observation': self.eval_env.observation_spec(),
165
+ # },
166
+ # # self.eval_env.observation_spec()['features'],
167
+ # self.eval_env.action_spec(),
168
+ # specs.Array(self.eval_env.action_spec().shape, self.eval_env.action_spec().dtype, 'vinn_action'),
169
+ # specs.Array((1, ), np.float32, 'reward'),
170
+ # specs.Array((1, ), np.float32, 'discount'),
171
+ # ]
172
+
173
+ # self.eval_replay_storage = ZarrEvalReplayBufferStorage(data_specs, self.work_dir / 'eval_buffer', debug_timestamps=self.cfg.debug_timestamps, save_buffer=self.cfg.save_buffer, debug_info_data_specs=self.eval_env.debug_info_data_specs, camera_info_dict=self.eval_env.get_camera_info_dict())
174
+ self.eval_replay_storage = RosbagEvalReplayBufferStorage(self.work_dir)
175
+
176
+ self.video_recorder = VideoRecorder(
177
+ self.work_dir if self.cfg.save_video else None,
178
+ ros_enabled=True,
179
+ fps=self.cfg.agent.config.policy_frequency,
180
+ )
181
+
182
+ print('workspace setup complete')
183
+
184
+ @property
185
+ def global_step(self):
186
+ # return self._global_step
187
+ return self.eval_env.get_global_step()
188
+
189
+ @property
190
+ def global_episode(self):
191
+ return self._global_episode
192
+
193
+ @property
194
+ def global_frame(self):
195
+ return self.global_step * self.cfg.action_repeat
196
+
197
+ @property
198
+ def global_epoch(self):
199
+ return self._global_epoch
200
+
201
+ def reset(self, eval_idx):
202
+ if not self.eval_env.enable_arm:
203
+ return np.array([0,0,0], dtype=np.float32)
204
+ self.eval_env.arm_refresh(reset=False)
205
+ # Set start position
206
+ try:
207
+ self.eval_env.set_position(self.start_pos[eval_idx])
208
+ except:
209
+ self.eval_env.arm.set_position(self.start_pos[eval_idx])
210
+ if self.eval_env.arm.keep_gripper_closed:
211
+ self.eval_env.arm.close_gripper_fully()
212
+ else:
213
+ self.eval_env.arm.open_gripper_fully()
214
+ time.sleep(0.1)
215
+ time_step = self.eval_env.step(np.zeros(self.eval_env.action_spec().shape[0], dtype=np.float32),
216
+ np.zeros(self.eval_env.action_spec().shape[0], dtype=np.float32))
217
+ return time_step
218
+
219
+ def eval(self):
220
+ # before evals start, prompt user for name of grasped object and the left book of the slot location
221
+ grasped_obj_name = input("Enter the name of the grasped object: ")
222
+ left_book_slot = input("Enter the left book slot location: ")
223
+ # update wandb config
224
+ if self.cfg.use_wandb:
225
+ wandb.config.update({'grasped_obj_name': grasped_obj_name, 'left_book_slot': left_book_slot})
226
+
227
+ self.preempt_episode = False
228
+ self.exit_eval = False
229
+ self.restart_episode = False
230
+
231
+ self.continue_keypress_thread = True
232
+ self.proceed_after_env_reset_event = threading.Event()
233
+ self.keypress_input_thread = threading.Thread(target=self.check_for_key_press)
234
+ self.keypress_input_thread.start()
235
+
236
+ # # Set model to eval mode
237
+ # self.agent.train(False)
238
+
239
+ eval_until_episode = utils.Until(self.cfg.num_eval)
240
+
241
+ self.use_action_history = False
242
+ # if "dp" in repr(self.agent) and "observation.action_history" in self.cfg.agent.config.policy_cfg.input_shapes:
243
+ if "observation.action_history" in self.cfg.agent.config.policy_cfg.input_shapes:
244
+ self.use_action_history = True
245
+
246
+ # self.eval_replay_storage._new_eval_step(0)
247
+
248
+ # if 'vinn' in repr(self.agent) or 'openloop' in repr(self.agent):
249
+ # with open(self.cfg.expert_dataset, 'rb') as f:
250
+ # if self.cfg.obs_type == 'pixels':
251
+ # self.expert_demo, _, self.expert_action, self.expert_reward = pickle.load(f)
252
+ # elif self.cfg.obs_type == 'features':
253
+ # _, self.expert_demo, self.expert_action, self.expert_reward = pickle.load(f)
254
+
255
+ # if self.cfg.action_trajectories:
256
+ # with open(self.cfg.expert_action_trajectories, 'rb') as f:
257
+ # self.expert_action = pickle.load(f)
258
+
259
+ # if isinstance(self.cfg.train_demo_idxs_list_or_num, int):
260
+ # if self.cfg.train_demo_idxs_list_or_num == -1:
261
+ # self.cfg.train_demo_idxs_list_or_num = len(self.expert_demo)
262
+ # train_demo_idxs_list_or_num = list(range(self.cfg.train_demo_idxs_list_or_num))
263
+
264
+ # self.expert_demo = self.expert_demo[train_demo_idxs_list_or_num]
265
+ # self.expert_action = self.expert_action[train_demo_idxs_list_or_num]
266
+ # self.expert_reward = self.expert_reward[train_demo_idxs_list_or_num]
267
+ # # if self.cfg.action_plans:
268
+ # # self.expert_action_plans = self.expert_action_plans[self.cfg.train_demo_idxs_list_or_num]
269
+ # # self.expert_demo = self.expert_demo[:self.cfg.num_demos]
270
+ # # self.expert_action = self.expert_action[:self.cfg.num_demos]
271
+ # # self.expert_reward = self.expert_reward[:self.cfg.num_demos]
272
+
273
+ # self.expert_demo = np.concatenate(self.expert_demo, axis=0)
274
+ # self.expert_rgb_obs = np.ascontiguousarray(np.transpose(self.expert_demo, (0,2,3,1))[:, :,:,:3].astype(np.uint8))
275
+ # self.expert_action = np.concatenate(self.expert_action, axis=0)
276
+
277
+ # self.agent.save_representations(self.expert_demo, self.expert_action, 128, config=self.expert_demo_config)
278
+
279
+ # Get start points
280
+ if self.cfg.random_start:
281
+ eval_starts = Path(self.cfg.eval_starts) / 'starts.pkl'
282
+ if eval_starts.exists():
283
+ with eval_starts.open('rb') as f:
284
+ self.start_pos = pickle.load(f)
285
+ else:
286
+ eval_starts = Path(self.cfg.eval_starts)
287
+ eval_starts.mkdir(parents=True, exist_ok=True)
288
+
289
+ # Generate start points
290
+ self.start_pos = []
291
+ try:
292
+ for _ in range(self.cfg.num_eval):
293
+ self.start_pos.append(self.eval_env.get_random_pos())
294
+ except:
295
+ for _ in range(self.cfg.num_eval):
296
+ self.start_pos.append(self.eval_env.arm.get_random_pos())
297
+
298
+ # Save start points for the task
299
+ eval_starts = eval_starts / 'starts.pkl'
300
+ with eval_starts.open('wb') as f:
301
+ pickle.dump(self.start_pos, f)
302
+
303
+ time_step = self.eval_env.reset()
304
+ # replay_thread = None
305
+ while eval_until_episode(self.global_episode) and not self.exit_eval:
306
+ # self.video_recorder.init(self.eval_env, video_filename=f'{self.global_episode}_eval.mp4')
307
+ print(f"Starting episode {self.global_episode}")
308
+ time_step = self.eval_env.reset() #Leon: need to call reset twice in case objects are trapped
309
+ self.video_recorder.init(self.eval_env, video_filename=f'{self.global_episode}_eval.mp4')
310
+ # x = input("Press Enter to continue... after reseting env")
311
+ print("Press Enter to continue... after reseting env. To rate prev episode, press 0 for failure and 1 for success")
312
+ self.proceed_after_env_reset_event.clear() # clear the event flag
313
+ self.proceed_after_env_reset_event.wait() # blocking wait for the event flag to be set
314
+ if self.global_episode > 0:
315
+ self.logger.log_metrics({'num_success': self.num_episode_successes}, self.global_step, 'eval', episode=self.global_episode)
316
+ self.logger.log_metrics({'success_rate': self.num_episode_successes/self.global_episode}, self.global_step, 'eval', episode=self.global_episode)
317
+ time_step = self.eval_env.reset()
318
+ # debug_info_dict = self.eval_env.debug_info_dict
319
+ # if replay_thread is not None:
320
+ # # wait for the last replay thread to finish
321
+ # replay_thread.join()
322
+
323
+ # self.eval_replay_storage.add(time_step._replace(observation=time_step.observation[self.cfg.obs_type]), debug_info_dict)
324
+ # replay_thread = threading.Thread(target=self.eval_replay_storage.add, args=(time_step._replace(observation=time_step.observation[self.cfg.obs_type]), debug_info_dict))
325
+ # replay_thread = threading.Thread(target=self.eval_replay_storage.add, args=(time_step, debug_info_dict))
326
+
327
+ # replay_thread.start()
328
+ if self.cfg.random_start:
329
+ time_step = self.reset(self.global_episode)
330
+ time.sleep(2) #5)
331
+ # if 'vinn' in repr(self.agent):
332
+ # self.agent.reset()
333
+ # # self.agent.buffer.reset()
334
+ # # if self.cfg.open_loop:
335
+ # # self.agent.current_step = 0
336
+ # if 'openloop' in repr(self.agent):
337
+ # self.agent.curr_step = 0
338
+ # at start of each episode, provide zero action for policies that use action history
339
+ # shape should be (T_o, T_a, action_dim)
340
+
341
+ # while not time_step.last() and not self.preempt_episode:
342
+ self.video_recorder.ros_start_recording()
343
+ self.eval_replay_storage.start_episode()
344
+ self.eval_env.start_policy_timer()
345
+ while not self.eval_env.episode_done() and not self.preempt_episode:
346
+ # with torch.no_grad(), utils.eval_mode(self.agent):
347
+ # # if self.cfg.agent.provide_topk:
348
+ # # action, vinn_action, topk = self.agent.act(
349
+ # # time_step.observation['pixels'],
350
+ # # self.global_step,
351
+ # # eval_mode=True)
352
+ # # elif self.cfg.agent.provide_obs:
353
+ # # action, vinn_action, obs = self.agent.act(
354
+ # # time_step.observation['pixels'],
355
+ # # self.global_step,
356
+ # # eval_mode=True)
357
+ # # else:
358
+ # action, vinn_action = self.agent.act(
359
+ # time_step.observation,
360
+ # self.global_step,
361
+ # eval_mode=True,
362
+ # obs_timestamp=time_step.observation['timestamp'],
363
+ # obs_seq=time_step.observation['seq'],
364
+ # action_history=action_history,
365
+ # action_history_start_timestamp=action_history_start_timestamp,
366
+ # )
367
+ # DONT WAIT FOR POLICY TO GET AN ACTION
368
+ # we dont want to slow down grabbing obs and passing to sam/contact features
369
+
370
+ self.eval_env.run_policy_threads() # this just does a rospy sleep
371
+
372
+ # if self.use_action_history:
373
+ # action_history_start_timestamp = time_step.observation['timestamp']
374
+ # # action_history = action[:self.cfg.agent.config.policy_cfg.action_history_encoder_config.history_length, ...]
375
+ # # add n_obs_steps dimension to action_history, for now we assume n_obs_steps = 1
376
+ # # TODO: handle n_obs_steps > 1
377
+ # action_history = action[np.newaxis, ...]
378
+
379
+ # time_step = self.eval_env.step(action, vinn_action) # obs, reward after action has been taken
380
+ # debug_info_dict = self.eval_env.debug_info_dict
381
+
382
+ # time_step = self.eval_env.ros_step()
383
+
384
+ # replay_thread.join()
385
+
386
+ # time how long it takes to execute the step
387
+ # time_before_add = time.perf_counter()
388
+ # self.eval_replay_storage.add(time_step._replace(observation=time_step.observation[self.cfg.obs_type]), debug_info_dict)
389
+ # use thread to call the add function in a separate thread
390
+ # replay_thread = threading.Thread(target=self.eval_replay_storage.add, args=(time_step._replace(observation=time_step.observation[self.cfg.obs_type]), debug_info_dict))
391
+
392
+ # replay_thread = threading.Thread(target=self.eval_replay_storage.add, args=(time_step, debug_info_dict))
393
+ # replay_thread.start()
394
+
395
+ # print(f"Time to add to replay buffer: {time.perf_counter() - time_before_add}")
396
+
397
+ # self.video_recorder.record(self.eval_env)
398
+ # self._global_step += 1
399
+
400
+ self.eval_env.stop_policy_timer()
401
+
402
+ if self.restart_episode:
403
+ # means we should delete the current episode and start again
404
+ self.restart_episode = False
405
+ self.eval_replay_storage.reset_current_episode()
406
+ self.video_recorder.reset_current_episode()
407
+
408
+ else:
409
+ self.eval_replay_storage.store_current_episode()
410
+ video_filepath = self.video_recorder.save()
411
+ self.logger.log_video(f"eval/{video_filepath.name.rstrip('.mp4')}", video_filepath, self.global_step)
412
+ self._global_episode += 1
413
+
414
+ self.preempt_episode = False # reset preempt_episode flag
415
+
416
+ # self.video_recorder.save(f'{episode}_eval.mp4')
417
+ # get the video file and convert to video tensor to log
418
+
419
+ self.eval_env.reset()
420
+
421
+ print("Evaluation finished. To wrap up, rate prev episode, press 0 for failure and 1 for success")
422
+ self.proceed_after_env_reset_event.clear() # clear the event flag
423
+ self.proceed_after_env_reset_event.wait() # blocking wait for the event flag to be set
424
+ if self.global_episode > 0:
425
+ # self.logger.log_metrics({'num_success': self.num_episode_successes}, self.global_step, 'eval', episode=self.global_episode)
426
+ self.logger.log_metrics({'num_success': self.num_episode_successes}, self.global_step, 'eval', episode=self.global_episode)
427
+ self.logger.log_metrics({'success_rate': self.num_episode_successes/self.global_episode}, self.global_step, 'eval', episode=self.global_episode)
428
+
429
+ self.continue_keypress_thread = False # will stop the keypress thread
430
+ self.keypress_input_thread.join() # wait for the keypress thread to finish
431
+
432
+ def load_checkpoint_conf(self, snapshot_path):
433
+ config_path = snapshot_path.parent / 'config.yaml'
434
+ if not config_path.exists():
435
+ raise FileNotFoundError(f'No snapshot conf found at {config_path}')
436
+ else:
437
+ # load the omegaconf config
438
+ hydra.core.global_hydra.GlobalHydra.instance().clear()
439
+ hydra.initialize(
440
+ str(_relative_path_between(Path(config_path).absolute().parent, Path(__file__).absolute().parent)),
441
+ )
442
+ cfg = hydra.compose(Path(config_path).stem)
443
+ from deepdiff import DeepDiff
444
+ from omegaconf import open_dict
445
+ diff = DeepDiff(OmegaConf.to_container(cfg), OmegaConf.to_container(self.cfg)) # old, new
446
+ # import re
447
+ overwriteable_keys = [f"root{overwritable_key}" for overwritable_key in ["['use_wandb']", "['path_to_depth_extrinsics']", "['eval']", "['root_dir']", "['wandb_notes']", "['agent']['config']['train_cfg']['use_amp']", "['agent']['config']['compile']", "['agent']['config']['policy_cfg']['num_inference_steps']"]]
448
+ if "values_changed" in diff:
449
+ # top_k_checkpoints, wandb_notes, agent.config.train_cfg.use_amp, save_snapshot_every_epochs_diffusion, check_topk_every_epochs_diffusion, validate_diffusion_on_action_loss_every_epochs, train_eval_diffusion_on_action_loss_every_epochs, validate_every_epochs_diffusion
450
+ # for keys above, overwrite the old config with the new config
451
+ for k, v in diff['values_changed'].items():
452
+ # replace any keys that are under "root['suite']"
453
+ if k in overwriteable_keys or k.startswith("root['suite']"):
454
+ print(f"Found changed key {k} with value {v}. Overwriting old checkpoint config")
455
+ if k == "root['agent']['config']['compile']":
456
+ if diff['values_changed'][k]['new_value']:
457
+ self.loading_uncompiled_checkpoint_with_compile = True
458
+ elif not diff['values_changed'][k]['new_value']:
459
+ # raise ValueError("Cannot load a compiled checkpoint without compile")
460
+ self.loading_compiled_checkpoint_with_no_compile = True
461
+ exec(f"{k.replace('root[', 'cfg[')} = {k.replace('root[', 'self.cfg[')}")
462
+ # for any new values, update the old checkpoint config
463
+ if "dictionary_item_added" in diff:
464
+ for new_key in diff['dictionary_item_added']: # this is a list
465
+ # if new_key == "root['suite']['task_make_fn']['observation_cfg']":
466
+ if new_key == "root['suite']['task_make_fn']['agent_policy_cfg']":
467
+ # pass the agents observation_cfg to the suite task_make_fn
468
+ with open_dict(cfg): # to allow addition of non-existing keys
469
+ # cfg.suite.task_make_fn.observation_cfg = cfg.agent.config.observation_cfg
470
+ cfg.suite.task_make_fn.agent_policy_cfg = cfg.agent.config
471
+ continue
472
+ elif "['agent']['config']['policy_cfg']['input_shapes']" in new_key:
473
+ # skip adding the new key if it is the input_shapes of the policy_cfg
474
+ continue
475
+ else:
476
+ print(f"Found new key {new_key} with value {eval(new_key.replace('root[', 'self.cfg['))}. Adding to checkpoint config")
477
+ # eval(new_key.replace('root', 'cfg')) = eval(new_key.replace('root', 'self.cfg'))
478
+ if new_key == "root['agent']['config']['compile']":
479
+ if self.cfg.agent.config.compile:
480
+ self.loading_uncompiled_checkpoint_with_compile = True
481
+
482
+ with open_dict(cfg):
483
+ exec(f"{new_key.replace('root[', 'cfg[')}={new_key.replace('root[', 'self.cfg[')}")
484
+ self.cfg = cfg
485
+
486
+ def load_checkpoint(self, snapshot_path, bc=False):
487
+ print(f'resuming {repr(self.agent)}: {snapshot_path}')
488
+ with snapshot_path.open('rb') as f:
489
+ payload = torch.load(f)
490
+ agent_payload = {}
491
+ for k, v in payload.items():
492
+ if k not in self.__dict__:
493
+ agent_payload[k] = v
494
+ elif k == '_global_epoch':
495
+ self._global_epoch = v
496
+ print(f'loaded epoch: {v}')
497
+ if self.cfg.use_wandb:
498
+ # add to config of wandb
499
+ wandb.config.update({'epoch': v})
500
+
501
+ # self.agent.load_snapshot_eval(agent_payload, bc)
502
+
503
+ @hydra.main(config_path='cfgs', config_name='config_eval')
504
+ def main(cfg):
505
+ from eval_robot import Workspace as W
506
+ root_dir = Path.cwd()
507
+ workspace = W(cfg)
508
+
509
+ workspace.eval()
510
+
511
+ if __name__ == '__main__':
512
+ main()
205123/wandb/run-20241210_205130-svysfc56/files/config.yaml ADDED
@@ -0,0 +1,966 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ wandb_version: 1
2
+
3
+ root_dir:
4
+ desc: null
5
+ value: /home/leonmkim/fish_leon
6
+ replay_buffer_size:
7
+ desc: null
8
+ value: 150000
9
+ replay_buffer_num_workers:
10
+ desc: null
11
+ value: 2
12
+ nstep:
13
+ desc: null
14
+ value: 3
15
+ batch_size:
16
+ desc: null
17
+ value: 128
18
+ seed:
19
+ desc: null
20
+ value: 0
21
+ dataset_shuffle_seed:
22
+ desc: null
23
+ value: 0
24
+ device:
25
+ desc: null
26
+ value: cuda
27
+ save_video:
28
+ desc: null
29
+ value: true
30
+ save_train_video:
31
+ desc: null
32
+ value: true
33
+ use_tb:
34
+ desc: null
35
+ value: true
36
+ use_wandb:
37
+ desc: null
38
+ value: true
39
+ wandb_run_id:
40
+ desc: null
41
+ value: '265626'
42
+ wandb_notes:
43
+ desc: null
44
+ value: 265626_req_265827
45
+ eval:
46
+ desc: null
47
+ value: true
48
+ process_contact_features:
49
+ desc: null
50
+ value: true
51
+ obs_type:
52
+ desc: null
53
+ value: pixels
54
+ use_color:
55
+ desc: null
56
+ value: true
57
+ use_depth:
58
+ desc: null
59
+ value: true
60
+ use_masks:
61
+ desc: null
62
+ value: true
63
+ mask_list:
64
+ desc: null
65
+ value:
66
+ - EE_obj_mask
67
+ mask_representation:
68
+ desc: null
69
+ value: channels
70
+ crop_hw:
71
+ desc: null
72
+ value:
73
+ - 144
74
+ - 144
75
+ crop_down_offset:
76
+ desc: null
77
+ value: 48
78
+ color_crop_type:
79
+ desc: null
80
+ value: null
81
+ depth_crop_type:
82
+ desc: null
83
+ value: null
84
+ segmask_crop_type:
85
+ desc: null
86
+ value: null
87
+ add_crop_binary_mask:
88
+ desc: null
89
+ value: false
90
+ add_coord_conv_map:
91
+ desc: null
92
+ value: false
93
+ use_context_color:
94
+ desc: null
95
+ value: false
96
+ use_context_depth:
97
+ desc: null
98
+ value: false
99
+ use_context_segmask:
100
+ desc: null
101
+ value: false
102
+ context_color_crop_type:
103
+ desc: null
104
+ value: null
105
+ context_depth_crop_type:
106
+ desc: null
107
+ value: null
108
+ context_segmask_crop_type:
109
+ desc: null
110
+ value: null
111
+ context_add_crop_binary_mask:
112
+ desc: null
113
+ value: false
114
+ context_add_coord_conv_map:
115
+ desc: null
116
+ value: false
117
+ use_contact_map:
118
+ desc: null
119
+ value: false
120
+ use_sdf_maps:
121
+ desc: null
122
+ value: false
123
+ use_normals_maps:
124
+ desc: null
125
+ value: false
126
+ which_objects:
127
+ desc: null
128
+ value: both
129
+ max_contact_prob:
130
+ desc: null
131
+ value: 0.1
132
+ max_depth:
133
+ desc: null
134
+ value: 2.0
135
+ grasped_dtc_max_value:
136
+ desc: null
137
+ value: 0.2
138
+ env_dtc_max_value:
139
+ desc: null
140
+ value: 0.4
141
+ grasped_normals_mask_max_dtc_value:
142
+ desc: null
143
+ value: 0.2
144
+ env_normals_mask_max_dtc_value:
145
+ desc: null
146
+ value: 0.4
147
+ clamp_dtc:
148
+ desc: null
149
+ value: true
150
+ dtc_adaptive_normalization:
151
+ desc: null
152
+ value: false
153
+ mask_normals_within_sdf:
154
+ desc: null
155
+ value: true
156
+ adaptive_normals_mask:
157
+ desc: null
158
+ value: true
159
+ learnable_contact_preprocess_params:
160
+ desc: null
161
+ value: true
162
+ contact_model_name:
163
+ desc: null
164
+ value: local_multitask_outhd64all_home_crop_h144w144d48_mask_ctxtmask_seed_220979_epoch_9
165
+ contact_estimation_model_ckpt_path:
166
+ desc: null
167
+ value: ~/fish_leon/contact_estimation/artifacts/175604_2/checkpoints/epoch=09-val_loss=0.00.ckpt
168
+ encoder_type:
169
+ desc: null
170
+ value: small
171
+ debug_timestamps:
172
+ desc: null
173
+ value: false
174
+ open_loop:
175
+ desc: null
176
+ value: false
177
+ action_trajectories:
178
+ desc: null
179
+ value: true
180
+ stop_after_action:
181
+ desc: null
182
+ value: false
183
+ interpolation_frequency:
184
+ desc: null
185
+ value: 25
186
+ policy_frequency:
187
+ desc: null
188
+ value: 5
189
+ wait_for_new_camera_frames:
190
+ desc: null
191
+ value: true
192
+ baseline:
193
+ desc: null
194
+ value: false
195
+ train_demo_idxs_list_or_num:
196
+ desc: null
197
+ value: -1
198
+ log_train_every_steps:
199
+ desc: null
200
+ value: 25
201
+ name_of_expert_demo:
202
+ desc: null
203
+ value: 64_240x320_hbm_twodim_fps_fix_contact_fix_recovery_fix_annotated_start_idx_5hz_zstd7_EE_pxl_coords_expert_demos_imp_act
204
+ expert_dataset_dirpath:
205
+ desc: null
206
+ value: /home/leonmkim/fish_leon/FISH/expert_demos/frankagym/FrankaInsertion-v1/64_240x320_hbm_twodim_fps_fix_contact_fix_recovery_fix_annotated_start_idx_5hz_zstd7_EE_pxl_coords_expert_demos_imp_act
207
+ store_dataset_in_memory:
208
+ desc: null
209
+ value: false
210
+ expert_dataset:
211
+ desc: null
212
+ value: /home/leonmkim/fish_leon/FISH/expert_demos/frankagym/FrankaInsertion-v1/64_240x320_hbm_twodim_fps_fix_contact_fix_recovery_fix_annotated_start_idx_5hz_zstd7_EE_pxl_coords_expert_demos_imp_act/demos.zarr
213
+ action_key:
214
+ desc: null
215
+ value: action_trajectory_25hz
216
+ semantic_demo_grouping_name:
217
+ desc: null
218
+ value: semantic_demo_grouping.yaml
219
+ semantic_demo_grouping:
220
+ desc: null
221
+ value: /home/leonmkim/fish_leon/FISH/expert_demos/frankagym/FrankaInsertion-v1/64_240x320_hbm_twodim_fps_fix_contact_fix_recovery_fix_annotated_start_idx_5hz_zstd7_EE_pxl_coords_expert_demos_imp_act/semantic_demo_grouping.yaml
222
+ include_groups_list:
223
+ desc: null
224
+ value:
225
+ - hbm_twodim_left_to_right
226
+ expert_dataset_config:
227
+ desc: null
228
+ value: /home/leonmkim/fish_leon/FISH/expert_demos/frankagym/FrankaInsertion-v1/64_240x320_hbm_twodim_fps_fix_contact_fix_recovery_fix_annotated_start_idx_5hz_zstd7_EE_pxl_coords_expert_demos_imp_act/demo_config.yaml
229
+ name_of_valid_demo:
230
+ desc: null
231
+ value: 64_240x320_hbm_twodim_fps_fix_contact_fix_recovery_fix_annotated_start_idx_5hz_zstd7_EE_pxl_coords_expert_demos_imp_act
232
+ valid_dataset_dir:
233
+ desc: null
234
+ value: /home/leonmkim/fish_leon/FISH/expert_demos/frankagym/FrankaInsertion-v1/64_240x320_hbm_twodim_fps_fix_contact_fix_recovery_fix_annotated_start_idx_5hz_zstd7_EE_pxl_coords_expert_demos_imp_act/demos.zarr
235
+ valid_demo_idxs_list_or_num:
236
+ desc: null
237
+ value: null
238
+ val_num_groups:
239
+ desc: null
240
+ value: 0
241
+ load_bc:
242
+ desc: null
243
+ value: true
244
+ checkpoint_epoch_list:
245
+ desc: null
246
+ value:
247
+ - 99
248
+ - 199
249
+ - 299
250
+ - 399
251
+ - 499
252
+ - 599
253
+ - 699
254
+ - 799
255
+ - 899
256
+ - 999
257
+ - 1249
258
+ - 1499
259
+ - 1749
260
+ - 1999
261
+ - 2999
262
+ - 3999
263
+ - 4999
264
+ - 5999
265
+ - 6999
266
+ - 7999
267
+ - 8999
268
+ - 9999
269
+ snapshot_root_dir:
270
+ desc: null
271
+ value: /mnt/grasp_high_usage/leonmkim/contact_estimation/FISH
272
+ save_snapshot:
273
+ desc: null
274
+ value: true
275
+ save_last_snapshot:
276
+ desc: null
277
+ value: true
278
+ save_snapshot_when_done:
279
+ desc: null
280
+ value: true
281
+ top_k_checkpoints:
282
+ desc: null
283
+ value: 5
284
+ save_snapshot_link_to_weights_dir:
285
+ desc: null
286
+ value: deprecated
287
+ bc_regularize:
288
+ desc: null
289
+ value: false
290
+ bc_weight_type:
291
+ desc: null
292
+ value: qfilter
293
+ experiment_dir:
294
+ desc: null
295
+ value: ./exp_local/frankagym_pixels/FrankaInsertion-v1/265626
296
+ agent:
297
+ desc: null
298
+ value:
299
+ _target_: agent.diffusion_policy.DiffusionPolicyAgent
300
+ name: diffusion_policy
301
+ load_checkpoint: true
302
+ obs_shape:
303
+ - 5
304
+ - 240
305
+ - 320
306
+ action_shape:
307
+ - 36
308
+ - 7
309
+ device: cuda
310
+ n_obs_steps: 1
311
+ suite_name: frankagym
312
+ obs_type: pixels
313
+ enable_arm: true
314
+ enable_camera: true
315
+ use_tb: true
316
+ desired_image_shape:
317
+ - 13
318
+ - 180
319
+ - 240
320
+ orig_cam_shape:
321
+ - 3
322
+ - 240
323
+ - 320
324
+ config:
325
+ _target_: agent.diffusion_policy.DiffusionPolicyAgentConfig
326
+ compile: false
327
+ device: cuda
328
+ cam_resize_shape:
329
+ - 13
330
+ - 180
331
+ - 240
332
+ orig_cam_shape:
333
+ - 3
334
+ - 240
335
+ - 320
336
+ policy_cfg:
337
+ _target_: lerobot.common.policies.diffusion.configuration_diffusion.DiffusionConfig
338
+ n_obs_steps: 1
339
+ horizon: 36
340
+ n_action_steps: 36
341
+ output_shapes:
342
+ action:
343
+ - 7
344
+ input_normalization_modes:
345
+ observation.image: mean_std
346
+ observation.state: min_max
347
+ observation.action_history: min_max
348
+ output_normalization_modes:
349
+ action: min_max
350
+ vision_backbone: resnet18
351
+ pretrained_backbone_weights: null
352
+ transforms:
353
+ - _target_: torchaug.transforms.RandomAffine
354
+ degrees:
355
+ - -5
356
+ - 5
357
+ translate:
358
+ - 0.05
359
+ - 0.05
360
+ batch_transform: true
361
+ num_chunks: -1
362
+ batch_inplace: true
363
+ - _target_: torchaug.transforms.RandomColorJitter
364
+ brightness: 0.3
365
+ contrast: 0.4
366
+ saturation: 0.5
367
+ hue: 0.08
368
+ batch_transform: true
369
+ num_chunks: -1
370
+ batch_inplace: true
371
+ use_group_norm: true
372
+ spatial_softmax_num_keypoints: 32
373
+ action_history_encoder_config:
374
+ _target_: lerobot.common.policies.diffusion.configuration_diffusion.Unet1dEncoderConfig
375
+ in_channels: 7
376
+ out_channels: 32
377
+ history_length: 6
378
+ kernel_size: 5
379
+ downsample_kernel_size: 3
380
+ downsample_stride: 2
381
+ downsample_padding: 1
382
+ down_dims:
383
+ - 256
384
+ - 512
385
+ - 1024
386
+ kernel_size: 5
387
+ n_groups: 8
388
+ diffusion_step_embed_dim: 128
389
+ use_film_scale_modulation: true
390
+ noise_scheduler_type: DDIM
391
+ beta_schedule: squaredcos_cap_v2
392
+ beta_start: 0.0001
393
+ beta_end: 0.02
394
+ prediction_type: epsilon
395
+ clip_sample: true
396
+ clip_sample_range: 1.0
397
+ num_train_timesteps: 50
398
+ num_inference_steps: 10
399
+ do_mask_loss_for_padding: false
400
+ input_shapes:
401
+ observation.image:
402
+ - 13
403
+ - 180
404
+ - 240
405
+ context_observation.image:
406
+ - 13
407
+ - 180
408
+ - 240
409
+ observation.state:
410
+ - 8
411
+ observation.action_history:
412
+ - 7
413
+ train_cfg:
414
+ _target_: utils.TrainConfig
415
+ lr: 0.0001
416
+ lr_scheduler: cosine
417
+ lr_warmup_steps: 500
418
+ adam_betas:
419
+ - 0.95
420
+ - 0.999
421
+ adam_eps: 1.0e-08
422
+ adam_weight_decay: 1.0e-06
423
+ grad_clip_norm: 10
424
+ offline_steps: 1000000
425
+ use_amp: true
426
+ observation_cfg:
427
+ _target_: agent.encoder.VisualFeatureSet
428
+ use_depth: true
429
+ use_color: true
430
+ mask_input_dict:
431
+ _target_: agent.encoder.MaskInputDict
432
+ enable: true
433
+ representation: channels
434
+ mask_list:
435
+ - EE_obj_mask
436
+ crop_input_config:
437
+ _target_: agent.encoder.CropInputConfig
438
+ color_crop_type: null
439
+ depth_crop_type: null
440
+ segmask_crop_type: null
441
+ crop_hw:
442
+ - 144
443
+ - 144
444
+ crop_down_offset: 48
445
+ add_crop_binary_mask: false
446
+ add_coord_conv_map: false
447
+ context_input_config:
448
+ _target_: agent.encoder.ContextInputConfig
449
+ use_color: false
450
+ use_depth: false
451
+ mask_input_dict:
452
+ _target_: agent.encoder.MaskInputDict
453
+ enable: false
454
+ representation: channels
455
+ mask_list:
456
+ - EE_obj_mask
457
+ crop_input_config:
458
+ _target_: agent.encoder.CropInputConfig
459
+ color_crop_type: null
460
+ depth_crop_type: null
461
+ segmask_crop_type: null
462
+ crop_hw:
463
+ - 144
464
+ - 144
465
+ crop_down_offset: 48
466
+ add_crop_binary_mask: false
467
+ add_coord_conv_map: false
468
+ mask_soft_approx_scheduler_config:
469
+ _target_: agent.encoder.MaskSoftApproxSchedulerConfig
470
+ num_steps: 40000
471
+ initial_value: 10.0
472
+ final_value: 1000.0
473
+ interpolation_scheme: cosine
474
+ use_contact_map: false
475
+ use_sdf_maps: false
476
+ use_normals_maps: false
477
+ which_objects: both
478
+ grasped_dtc_max_value: 0.2
479
+ env_dtc_max_value: 0.4
480
+ grasped_normals_mask_max_dtc_value: 0.2
481
+ env_normals_mask_max_dtc_value: 0.4
482
+ clamp_dtc: true
483
+ max_contact_prob: 0.1
484
+ mask_normals_within_sdf: true
485
+ dtc_adaptive_normalization: false
486
+ adaptive_normals_mask: true
487
+ max_depth: 2.0
488
+ image_shape:
489
+ - 13
490
+ - 180
491
+ - 240
492
+ learnable_contact_preprocess_params: true
493
+ learning_rate: 0.0001
494
+ weight_decay: 0.0
495
+ contact_model_name: local_multitask_outhd64all_home_crop_h144w144d48_mask_ctxtmask_seed_220979_epoch_9
496
+ zero_centered: false
497
+ policy_frequency: 5
498
+ interpolation_frequency: 25
499
+ suite:
500
+ desc: null
501
+ value:
502
+ suite: frankagym
503
+ name: frankagym
504
+ frame_stack: 1
505
+ action_repeat: 1
506
+ discount: 0.99
507
+ hidden_dim: 1024
508
+ num_train_frames: 2010
509
+ num_seed_frames: 260
510
+ num_train_epochs: 5000
511
+ validate_every_epochs: 100
512
+ validate_diffusion_on_action_loss_every_epochs: 500
513
+ train_eval_diffusion_on_action_loss_every_epochs: 500
514
+ check_topk_every_epochs: 10
515
+ save_snapshot_every_epochs: 5000
516
+ eval_every_frames: 2000
517
+ num_eval_episodes: 5
518
+ save_snapshot: true
519
+ wait_for_user_to_start_episode: true
520
+ task_make_fn:
521
+ _target_: suite.frankagym.make
522
+ name: FrankaInsertion-v1
523
+ height: 240
524
+ width: 320
525
+ frame_stack: 1
526
+ action_repeat: 1
527
+ seed: 0
528
+ enable_arm: true
529
+ enable_gripper: true
530
+ start_with_gripper_open: true
531
+ enable_camera: true
532
+ path_to_depth_extrinsics: ~/fish_leon/FISH/cfgs/camera_poses/camera_poses_L515/20240904-122305/color_tf_world.npy
533
+ contact_estimation_model_ckpt_path: ~/fish_leon/contact_estimation/artifacts/175604_2/checkpoints/epoch=09-val_loss=0.00.ckpt
534
+ x_limit:
535
+ - 0.2
536
+ - 0.7
537
+ y_limit:
538
+ - -0.4
539
+ - 0.4
540
+ z_limit:
541
+ - -0.05
542
+ - 0.55
543
+ device: cuda
544
+ interpolation_frequency: 25
545
+ policy_frequency: 5
546
+ debug_timestamps: false
547
+ stop_after_action: false
548
+ open_loop: false
549
+ wait_for_new_camera_frames: true
550
+ action_key: action_trajectory_25hz
551
+ action_trajectory_horizon: 36
552
+ action_trajectories: true
553
+ path_to_zarr_dataset: /home/leonmkim/fish_leon/FISH/expert_demos/frankagym/FrankaInsertion-v1/64_240x320_hbm_twodim_fps_fix_contact_fix_recovery_fix_annotated_start_idx_5hz_zstd7_EE_pxl_coords_expert_demos_imp_act/demos.zarr
554
+ observation_cfg:
555
+ _target_: agent.encoder.VisualFeatureSet
556
+ use_depth: true
557
+ use_color: true
558
+ mask_input_dict:
559
+ _target_: agent.encoder.MaskInputDict
560
+ enable: true
561
+ representation: channels
562
+ mask_list:
563
+ - EE_obj_mask
564
+ crop_input_config:
565
+ _target_: agent.encoder.CropInputConfig
566
+ color_crop_type: null
567
+ depth_crop_type: null
568
+ segmask_crop_type: null
569
+ crop_hw:
570
+ - 144
571
+ - 144
572
+ crop_down_offset: 48
573
+ add_crop_binary_mask: false
574
+ add_coord_conv_map: false
575
+ context_input_config:
576
+ _target_: agent.encoder.ContextInputConfig
577
+ use_color: false
578
+ use_depth: false
579
+ mask_input_dict:
580
+ _target_: agent.encoder.MaskInputDict
581
+ enable: false
582
+ representation: channels
583
+ mask_list:
584
+ - EE_obj_mask
585
+ crop_input_config:
586
+ _target_: agent.encoder.CropInputConfig
587
+ color_crop_type: null
588
+ depth_crop_type: null
589
+ segmask_crop_type: null
590
+ crop_hw:
591
+ - 144
592
+ - 144
593
+ crop_down_offset: 48
594
+ add_crop_binary_mask: false
595
+ add_coord_conv_map: false
596
+ mask_soft_approx_scheduler_config:
597
+ _target_: agent.encoder.MaskSoftApproxSchedulerConfig
598
+ num_steps: 40000
599
+ initial_value: 10.0
600
+ final_value: 1000.0
601
+ interpolation_scheme: cosine
602
+ use_contact_map: false
603
+ use_sdf_maps: false
604
+ use_normals_maps: false
605
+ which_objects: both
606
+ grasped_dtc_max_value: 0.2
607
+ env_dtc_max_value: 0.4
608
+ grasped_normals_mask_max_dtc_value: 0.2
609
+ env_normals_mask_max_dtc_value: 0.4
610
+ clamp_dtc: true
611
+ max_contact_prob: 0.1
612
+ mask_normals_within_sdf: true
613
+ dtc_adaptive_normalization: false
614
+ adaptive_normals_mask: true
615
+ max_depth: 2.0
616
+ image_shape:
617
+ - 13
618
+ - 180
619
+ - 240
620
+ learnable_contact_preprocess_params: true
621
+ learning_rate: 0.0001
622
+ weight_decay: 0.0
623
+ contact_model_name: local_multitask_outhd64all_home_crop_h144w144d48_mask_ctxtmask_seed_220979_epoch_9
624
+ zero_centered: false
625
+ agent_policy_cfg:
626
+ _target_: agent.diffusion_policy.DiffusionPolicyAgentConfig
627
+ compile: false
628
+ device: cuda
629
+ cam_resize_shape:
630
+ - 13
631
+ - 180
632
+ - 240
633
+ orig_cam_shape:
634
+ - 3
635
+ - 240
636
+ - 320
637
+ policy_cfg:
638
+ _target_: lerobot.common.policies.diffusion.configuration_diffusion.DiffusionConfig
639
+ n_obs_steps: 1
640
+ horizon: 36
641
+ n_action_steps: 36
642
+ output_shapes:
643
+ action:
644
+ - 7
645
+ input_normalization_modes:
646
+ observation.image: mean_std
647
+ observation.state: min_max
648
+ observation.action_history: min_max
649
+ output_normalization_modes:
650
+ action: min_max
651
+ vision_backbone: resnet18
652
+ pretrained_backbone_weights: null
653
+ transforms:
654
+ - _target_: torchaug.transforms.RandomAffine
655
+ degrees:
656
+ - -5
657
+ - 5
658
+ translate:
659
+ - 0.05
660
+ - 0.05
661
+ batch_transform: true
662
+ num_chunks: -1
663
+ batch_inplace: true
664
+ - _target_: torchaug.transforms.RandomColorJitter
665
+ brightness: 0.3
666
+ contrast: 0.4
667
+ saturation: 0.5
668
+ hue: 0.08
669
+ batch_transform: true
670
+ num_chunks: -1
671
+ batch_inplace: true
672
+ use_group_norm: true
673
+ spatial_softmax_num_keypoints: 32
674
+ action_history_encoder_config:
675
+ _target_: lerobot.common.policies.diffusion.configuration_diffusion.Unet1dEncoderConfig
676
+ in_channels: 7
677
+ out_channels: 32
678
+ history_length: 6
679
+ kernel_size: 5
680
+ downsample_kernel_size: 3
681
+ downsample_stride: 2
682
+ downsample_padding: 1
683
+ down_dims:
684
+ - 256
685
+ - 512
686
+ - 1024
687
+ kernel_size: 5
688
+ n_groups: 8
689
+ diffusion_step_embed_dim: 128
690
+ use_film_scale_modulation: true
691
+ noise_scheduler_type: DDIM
692
+ beta_schedule: squaredcos_cap_v2
693
+ beta_start: 0.0001
694
+ beta_end: 0.02
695
+ prediction_type: epsilon
696
+ clip_sample: true
697
+ clip_sample_range: 1.0
698
+ num_train_timesteps: 50
699
+ num_inference_steps: 10
700
+ do_mask_loss_for_padding: false
701
+ input_shapes:
702
+ observation.image:
703
+ - 13
704
+ - 180
705
+ - 240
706
+ context_observation.image:
707
+ - 13
708
+ - 180
709
+ - 240
710
+ observation.state:
711
+ - 8
712
+ observation.action_history:
713
+ - 7
714
+ train_cfg:
715
+ _target_: utils.TrainConfig
716
+ lr: 0.0001
717
+ lr_scheduler: cosine
718
+ lr_warmup_steps: 500
719
+ adam_betas:
720
+ - 0.95
721
+ - 0.999
722
+ adam_eps: 1.0e-08
723
+ adam_weight_decay: 1.0e-06
724
+ grad_clip_norm: 10
725
+ offline_steps: 1000000
726
+ use_amp: true
727
+ observation_cfg:
728
+ _target_: agent.encoder.VisualFeatureSet
729
+ use_depth: true
730
+ use_color: true
731
+ mask_input_dict:
732
+ _target_: agent.encoder.MaskInputDict
733
+ enable: true
734
+ representation: channels
735
+ mask_list:
736
+ - EE_obj_mask
737
+ crop_input_config:
738
+ _target_: agent.encoder.CropInputConfig
739
+ color_crop_type: null
740
+ depth_crop_type: null
741
+ segmask_crop_type: null
742
+ crop_hw:
743
+ - 144
744
+ - 144
745
+ crop_down_offset: 48
746
+ add_crop_binary_mask: false
747
+ add_coord_conv_map: false
748
+ context_input_config:
749
+ _target_: agent.encoder.ContextInputConfig
750
+ use_color: false
751
+ use_depth: false
752
+ mask_input_dict:
753
+ _target_: agent.encoder.MaskInputDict
754
+ enable: false
755
+ representation: channels
756
+ mask_list:
757
+ - EE_obj_mask
758
+ crop_input_config:
759
+ _target_: agent.encoder.CropInputConfig
760
+ color_crop_type: null
761
+ depth_crop_type: null
762
+ segmask_crop_type: null
763
+ crop_hw:
764
+ - 144
765
+ - 144
766
+ crop_down_offset: 48
767
+ add_crop_binary_mask: false
768
+ add_coord_conv_map: false
769
+ mask_soft_approx_scheduler_config:
770
+ _target_: agent.encoder.MaskSoftApproxSchedulerConfig
771
+ num_steps: 40000
772
+ initial_value: 10.0
773
+ final_value: 1000.0
774
+ interpolation_scheme: cosine
775
+ use_contact_map: false
776
+ use_sdf_maps: false
777
+ use_normals_maps: false
778
+ which_objects: both
779
+ grasped_dtc_max_value: 0.2
780
+ env_dtc_max_value: 0.4
781
+ grasped_normals_mask_max_dtc_value: 0.2
782
+ env_normals_mask_max_dtc_value: 0.4
783
+ clamp_dtc: true
784
+ max_contact_prob: 0.1
785
+ mask_normals_within_sdf: true
786
+ dtc_adaptive_normalization: false
787
+ adaptive_normals_mask: true
788
+ max_depth: 2.0
789
+ image_shape:
790
+ - 13
791
+ - 180
792
+ - 240
793
+ learnable_contact_preprocess_params: true
794
+ learning_rate: 0.0001
795
+ weight_decay: 0.0
796
+ contact_model_name: local_multitask_outhd64all_home_crop_h144w144d48_mask_ctxtmask_seed_220979_epoch_9
797
+ zero_centered: false
798
+ policy_frequency: 5
799
+ interpolation_frequency: 25
800
+ true_action_history: true
801
+ num_train_frames_bc:
802
+ desc: null
803
+ value: 50000
804
+ num_train_frames_drq:
805
+ desc: null
806
+ value: 1100000
807
+ stddev_schedule_drq:
808
+ desc: null
809
+ value: linear(1.0,0.1,100000)
810
+ task_name:
811
+ desc: null
812
+ value: FrankaInsertion-v1
813
+ num_train_frames_vinn:
814
+ desc: null
815
+ value: 25000
816
+ num_train_frames_diffusion:
817
+ desc: null
818
+ value: 1000000
819
+ num_train_epochs_bc:
820
+ desc: null
821
+ value: 5000
822
+ num_train_epochs_diffusion:
823
+ desc: null
824
+ value: 15000
825
+ validate_every_epochs_bc:
826
+ desc: null
827
+ value: 5
828
+ validate_every_epochs_diffusion:
829
+ desc: null
830
+ value: 250
831
+ validate_diffusion_on_action_loss_every_epochs:
832
+ desc: null
833
+ value: 250
834
+ train_eval_diffusion_on_action_loss_every_epochs:
835
+ desc: null
836
+ value: 250
837
+ check_topk_every_epochs:
838
+ desc: null
839
+ value: 5
840
+ check_topk_every_epochs_diffusion:
841
+ desc: null
842
+ value: 250
843
+ save_snapshot_every_epochs_diffusion:
844
+ desc: null
845
+ value: 1500
846
+ x_limit:
847
+ desc: null
848
+ value:
849
+ - 0.2
850
+ - 0.7
851
+ y_limit:
852
+ desc: null
853
+ value:
854
+ - -0.4
855
+ - 0.4
856
+ z_limit:
857
+ desc: null
858
+ value:
859
+ - -0.05
860
+ - 0.55
861
+ home_displacement:
862
+ desc: null
863
+ value:
864
+ - 0.55
865
+ - 0.0
866
+ - 0.55
867
+ - 180.0
868
+ - 0.0
869
+ - 0.0
870
+ enable_gripper:
871
+ desc: null
872
+ value: true
873
+ start_with_gripper_open:
874
+ desc: null
875
+ value: true
876
+ offset_mask:
877
+ desc: null
878
+ value:
879
+ - 1
880
+ - 1
881
+ - 1
882
+ - 1
883
+ - 1
884
+ - 1
885
+ path_to_depth_extrinsics:
886
+ desc: null
887
+ value: ~/fish_leon/FISH/cfgs/camera_poses/camera_poses_L515/20240904-122305/color_tf_world.npy
888
+ feature_type:
889
+ desc: null
890
+ value: 180x240_1_RGB_D_2.0_msk_channels_EE_obj_mask_acthst_hst6_out32_dwnkrnl3_dwnstrd2_dwnpd1
891
+ save_buffer:
892
+ desc: null
893
+ value: true
894
+ num_eval:
895
+ desc: null
896
+ value: 5
897
+ random_start:
898
+ desc: null
899
+ value: false
900
+ eval_starts:
901
+ desc: null
902
+ value: /home/leonmkim/fish_leon/FISH/eval_starts/frankagym_pixels/FrankaInsertion-v1
903
+ num_valid_demos:
904
+ desc: null
905
+ value: null
906
+ load_checkpoint:
907
+ desc: null
908
+ value: true
909
+ true_action_history:
910
+ desc: null
911
+ value: true
912
+ checkpoint_epoch:
913
+ desc: null
914
+ value: 12000
915
+ load_residual_weight:
916
+ desc: null
917
+ value: false
918
+ checkpoint_root_dir:
919
+ desc: null
920
+ value: /home/leonmkim/fish_leon/FISH
921
+ checkpoint_weight_dir:
922
+ desc: null
923
+ value: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626
924
+ residual_weight:
925
+ desc: null
926
+ value: /home/leonmkim/fish_leon/FISH/weights/frankagym_pixels/FrankaInsertion-v1/weight.pt
927
+ _wandb:
928
+ desc: null
929
+ value:
930
+ code_path: code/FISH/eval_robot.py
931
+ python_version: 3.10.14
932
+ cli_version: 0.17.5
933
+ framework: torch
934
+ is_jupyter_run: false
935
+ is_kaggle_kernel: false
936
+ start_time: 1733881890
937
+ t:
938
+ 1:
939
+ - 1
940
+ - 41
941
+ - 49
942
+ - 50
943
+ - 55
944
+ - 83
945
+ 2:
946
+ - 1
947
+ - 41
948
+ - 49
949
+ - 50
950
+ - 55
951
+ - 83
952
+ 3:
953
+ - 16
954
+ - 23
955
+ - 35
956
+ 4: 3.10.14
957
+ 5: 0.17.5
958
+ 8:
959
+ - 5
960
+ 13: linux-x86_64
961
+ grasped_obj_name:
962
+ desc: null
963
+ value: hbm
964
+ left_book_slot:
965
+ desc: null
966
+ value: twodim
205123/wandb/run-20241210_205130-svysfc56/files/diff.patch ADDED
@@ -0,0 +1,25 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ diff --git a/FISH/cfgs/config_eval.yaml b/FISH/cfgs/config_eval.yaml
2
+ index a9568e3..c42ff82 100644
3
+ --- a/FISH/cfgs/config_eval.yaml
4
+ +++ b/FISH/cfgs/config_eval.yaml
5
+ @@ -123,6 +123,11 @@ bc_weight_type: 'qfilter' # linear, qfilter
6
+
7
+ # Load weights
8
+ load_checkpoint: ${agent.load_checkpoint}
9
+ +# 64_240x320_hbm_twodim_fps_fix_contact_fix_recovery_fix_annotated_start_idx_5hz_zstd7_EE_pxl_coords_expert_demos_imp_act
10
+ +# full dataset
11
+ +wandb_run_id: "265626" # RGBD+Mask+act history
12
+ +# wandb_run_id: "265628" # RGBD+Mask+contact+act history
13
+ +
14
+ # 54_240x320_hbm_twodim_fps_fix_annotated_start_idx_5hz_zstd7_EE_pxl_coords_expert_demos_imp_act
15
+ # full dataset
16
+ # 6 step action history
17
+ @@ -135,7 +140,7 @@ load_checkpoint: ${agent.load_checkpoint}
18
+ # wandb_run_id: '264772_0' # RGBD+act history
19
+ # wandb_run_id: '264775_0' # RGBD+mask+act history
20
+ # wandb_run_id: '264776_0' # RGBD+contact+act history
21
+ -wandb_run_id: '264777_0' # RGBD+mask+contact+act history
22
+ +# wandb_run_id: '264777_0' # RGBD+mask+contact+act history
23
+
24
+ # 54_240x320_hbm_twodim_fps_fix_annotated_start_idx_5hz_zstd7_EE_pxl_coords_expert_demos_imp_act
25
+ # 16 held out demos
205123/wandb/run-20241210_205130-svysfc56/files/media/videos/eval/video_0_b6b0b3ba17b7513a5df9.mp4 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b6b0b3ba17b7513a5df9bca8084b0eda15d2f82df085783449312a47400bba10
3
+ size 942048
205123/wandb/run-20241210_205130-svysfc56/files/output.log ADDED
@@ -0,0 +1,1049 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+
2
+ loaded agent with feature_type: 180x240_1_RGB_D_2.0_msk_channels_EE_obj_mask_acthst_hst6_out32_dwnkrnl3_dwnstrd2_dwnpd1
3
+ [INFO] [1733881906.908934]: resetting environment
4
+ [INFO] [1733881906.913976]: cleared current plan
5
+ [INFO] [1733881906.915491]: moving to home
6
+ [INFO] [1733881908.619161]: reached home
7
+ [INFO] [1733881908.620458]: reset action history
8
+ [INFO] [1733881911.334703]: environment reset
9
+ Starting episode 0
10
+ [INFO] [1733881911.338240]: resetting environment
11
+ [INFO] [1733881911.346125]: cleared current plan
12
+ [INFO] [1733881911.347839]: moving to home
13
+ [INFO] [1733881912.349890]: reached home
14
+ [INFO] [1733881912.351231]: reset action history
15
+ [INFO] [1733881915.058954]: environment reset
16
+ Press Enter to continue... after reseting env. To rate prev episode, press 0 for failure and 1 for success
17
+ proceeding to start episode!
18
+ [INFO] [1733881926.140345]: resetting environment
19
+ [INFO] [1733881926.145656]: cleared current plan
20
+ [INFO] [1733881926.150768]: moving to home
21
+ [INFO] [1733881927.153098]: reached home
22
+ [INFO] [1733881927.154883]: reset action history
23
+ [INFO] [1733881929.863341]: environment reset
24
+ bag_path_name:=/home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/episode_rosbags/episode_0_2024-12-10-20-52-09.bag
25
+ ... logging to /home/leonmkim/.ros/log/1408e5ba-b74f-11ef-9a34-5546d6409a19/roslaunch-leonmkim-ROG-Strix-G15CS-G15CS-596892.log
26
+ started roslaunch server http://158.130.50.37:44627/
27
+ SUMMARY
28
+ ========
29
+ PARAMETERS
30
+ * /rosdistro: noetic
31
+ * /rosversion: 1.16.0
32
+ NODES
33
+ /
34
+ print_text (rostopic/rostopic)
35
+ pub_text (rostopic/rostopic)
36
+ rosbag_record (rosbag/record)
37
+ ROS_MASTER_URI=http://localhost:11311
38
+ process[pub_text-1]: started with pid [597118]
39
+ process[print_text-2]: started with pid [597119]
40
+ process[rosbag_record-3]: started with pid [597143]
41
+ started bagging!
42
+ [INFO] [1733881930.601407]: received mask_obs with header seq: 1
43
+ stamp:
44
+ secs: 1733881930
45
+ nsecs: 598197937
46
+ frame_id: '' and grasped_object_mask header seq: 7743
47
+ stamp:
48
+ secs: 1733881930
49
+ nsecs: 373498678
50
+ frame_id: "camera_color_optical_frame"
51
+ [INFO] [1733881930.790134]: received mask_obs with header seq: 2
52
+ stamp:
53
+ secs: 1733881930
54
+ nsecs: 787627696
55
+ frame_id: '' and grasped_object_mask header seq: 7749
56
+ stamp:
57
+ secs: 1733881930
58
+ nsecs: 573334217
59
+ frame_id: "camera_color_optical_frame"
60
+ [INFO] [1733881930.989544]: received mask_obs with header seq: 3
61
+ stamp:
62
+ secs: 1733881930
63
+ nsecs: 987669944
64
+ frame_id: '' and grasped_object_mask header seq: 7755
65
+ stamp:
66
+ secs: 1733881930
67
+ nsecs: 773170948
68
+ frame_id: "camera_color_optical_frame"
69
+ [INFO] [1733881931.186965]: received mask_obs with header seq: 4
70
+ stamp:
71
+ secs: 1733881931
72
+ nsecs: 185687780
73
+ frame_id: '' and grasped_object_mask header seq: 7761
74
+ stamp:
75
+ secs: 1733881930
76
+ nsecs: 973006487
77
+ frame_id: "camera_color_optical_frame"
78
+ [INFO] [1733881931.405436]: received mask_obs with header seq: 5
79
+ stamp:
80
+ secs: 1733881931
81
+ nsecs: 404456853
82
+ frame_id: '' and grasped_object_mask header seq: 7767
83
+ stamp:
84
+ secs: 1733881931
85
+ nsecs: 172842264
86
+ frame_id: "camera_color_optical_frame"
87
+ [INFO] [1733881931.606265]: received mask_obs with header seq: 6
88
+ stamp:
89
+ secs: 1733881931
90
+ nsecs: 605320453
91
+ frame_id: '' and grasped_object_mask header seq: 7773
92
+ stamp:
93
+ secs: 1733881931
94
+ nsecs: 372677803
95
+ frame_id: "camera_color_optical_frame"
96
+ [INFO] [1733881931.810085]: received mask_obs with header seq: 7
97
+ stamp:
98
+ secs: 1733881931
99
+ nsecs: 809148073
100
+ frame_id: '' and grasped_object_mask header seq: 7779
101
+ stamp:
102
+ secs: 1733881931
103
+ nsecs: 572512865
104
+ frame_id: "camera_color_optical_frame"
105
+ [INFO] [1733881932.000967]: received mask_obs with header seq: 8
106
+ stamp:
107
+ secs: 1733881932
108
+ nsecs: 77486
109
+ frame_id: '' and grasped_object_mask header seq: 7785
110
+ stamp:
111
+ secs: 1733881931
112
+ nsecs: 772349596
113
+ frame_id: "camera_color_optical_frame"
114
+ [INFO] [1733881932.200750]: received mask_obs with header seq: 9
115
+ stamp:
116
+ secs: 1733881932
117
+ nsecs: 199798583
118
+ frame_id: '' and grasped_object_mask header seq: 7791
119
+ stamp:
120
+ secs: 1733881931
121
+ nsecs: 972187757
122
+ frame_id: "camera_color_optical_frame"
123
+ [INFO] [1733881932.404178]: received mask_obs with header seq: 10
124
+ stamp:
125
+ secs: 1733881932
126
+ nsecs: 403140544
127
+ frame_id: '' and grasped_object_mask header seq: 7797
128
+ stamp:
129
+ secs: 1733881932
130
+ nsecs: 172024965
131
+ frame_id: "camera_color_optical_frame"
132
+ [INFO] [1733881932.603047]: received mask_obs with header seq: 11
133
+ stamp:
134
+ secs: 1733881932
135
+ nsecs: 602175712
136
+ frame_id: '' and grasped_object_mask header seq: 7803
137
+ stamp:
138
+ secs: 1733881932
139
+ nsecs: 371861935
140
+ frame_id: "camera_color_optical_frame"
141
+ [INFO] [1733881932.802278]: received mask_obs with header seq: 12
142
+ stamp:
143
+ secs: 1733881932
144
+ nsecs: 801258563
145
+ frame_id: '' and grasped_object_mask header seq: 7809
146
+ stamp:
147
+ secs: 1733881932
148
+ nsecs: 571698666
149
+ frame_id: "camera_color_optical_frame"
150
+ [INFO] [1733881933.018084]: received mask_obs with header seq: 13
151
+ stamp:
152
+ secs: 1733881933
153
+ nsecs: 17174005
154
+ frame_id: '' and grasped_object_mask header seq: 7815
155
+ stamp:
156
+ secs: 1733881932
157
+ nsecs: 771533251
158
+ frame_id: "camera_color_optical_frame"
159
+ [INFO] [1733881933.218519]: received mask_obs with header seq: 14
160
+ stamp:
161
+ secs: 1733881933
162
+ nsecs: 217555284
163
+ frame_id: '' and grasped_object_mask header seq: 7821
164
+ stamp:
165
+ secs: 1733881932
166
+ nsecs: 971366405
167
+ frame_id: "camera_color_optical_frame"
168
+ [INFO] [1733881933.411083]: received mask_obs with header seq: 15
169
+ stamp:
170
+ secs: 1733881933
171
+ nsecs: 410149812
172
+ frame_id: '' and grasped_object_mask header seq: 7827
173
+ stamp:
174
+ secs: 1733881933
175
+ nsecs: 171200514
176
+ frame_id: "camera_color_optical_frame"
177
+ [INFO] [1733881933.616644]: received mask_obs with header seq: 16
178
+ stamp:
179
+ secs: 1733881933
180
+ nsecs: 615602254
181
+ frame_id: '' and grasped_object_mask header seq: 7833
182
+ stamp:
183
+ secs: 1733881933
184
+ nsecs: 371033192
185
+ frame_id: "camera_color_optical_frame"
186
+ [INFO] [1733881933.810750]: received mask_obs with header seq: 17
187
+ stamp:
188
+ secs: 1733881933
189
+ nsecs: 809497833
190
+ frame_id: '' and grasped_object_mask header seq: 7839
191
+ stamp:
192
+ secs: 1733881933
193
+ nsecs: 570865870
194
+ frame_id: "camera_color_optical_frame"
195
+ [INFO] [1733881934.011081]: received mask_obs with header seq: 18
196
+ stamp:
197
+ secs: 1733881934
198
+ nsecs: 10175704
199
+ frame_id: '' and grasped_object_mask header seq: 7845
200
+ stamp:
201
+ secs: 1733881933
202
+ nsecs: 770703554
203
+ frame_id: "camera_color_optical_frame"
204
+ [INFO] [1733881934.204773]: received mask_obs with header seq: 19
205
+ stamp:
206
+ secs: 1733881934
207
+ nsecs: 203893184
208
+ frame_id: '' and grasped_object_mask header seq: 7851
209
+ stamp:
210
+ secs: 1733881933
211
+ nsecs: 970544338
212
+ frame_id: "camera_color_optical_frame"
213
+ [INFO] [1733881934.433582]: received mask_obs with header seq: 20
214
+ stamp:
215
+ secs: 1733881934
216
+ nsecs: 432651519
217
+ frame_id: '' and grasped_object_mask header seq: 7857
218
+ stamp:
219
+ secs: 1733881934
220
+ nsecs: 170385599
221
+ frame_id: "camera_color_optical_frame"
222
+ [INFO] [1733881934.628506]: received mask_obs with header seq: 21
223
+ stamp:
224
+ secs: 1733881934
225
+ nsecs: 626872062
226
+ frame_id: '' and grasped_object_mask header seq: 7864
227
+ stamp:
228
+ secs: 1733881934
229
+ nsecs: 403534174
230
+ frame_id: "camera_color_optical_frame"
231
+ [INFO] [1733881934.826157]: received mask_obs with header seq: 22
232
+ stamp:
233
+ secs: 1733881934
234
+ nsecs: 825274705
235
+ frame_id: '' and grasped_object_mask header seq: 7870
236
+ stamp:
237
+ secs: 1733881934
238
+ nsecs: 603375435
239
+ frame_id: "camera_color_optical_frame"
240
+ [INFO] [1733881935.031275]: received mask_obs with header seq: 23
241
+ stamp:
242
+ secs: 1733881935
243
+ nsecs: 29773235
244
+ frame_id: '' and grasped_object_mask header seq: 7876
245
+ stamp:
246
+ secs: 1733881934
247
+ nsecs: 803215981
248
+ frame_id: "camera_color_optical_frame"
249
+ [INFO] [1733881935.228458]: received mask_obs with header seq: 24
250
+ stamp:
251
+ secs: 1733881935
252
+ nsecs: 227554321
253
+ frame_id: '' and grasped_object_mask header seq: 7882
254
+ stamp:
255
+ secs: 1733881935
256
+ nsecs: 3057957
257
+ frame_id: "camera_color_optical_frame"
258
+ [INFO] [1733881935.422727]: received mask_obs with header seq: 25
259
+ stamp:
260
+ secs: 1733881935
261
+ nsecs: 421752691
262
+ frame_id: '' and grasped_object_mask header seq: 7887
263
+ stamp:
264
+ secs: 1733881935
265
+ nsecs: 169591427
266
+ frame_id: "camera_color_optical_frame"
267
+ [INFO] [1733881935.638195]: received mask_obs with header seq: 26
268
+ stamp:
269
+ secs: 1733881935
270
+ nsecs: 637306213
271
+ frame_id: '' and grasped_object_mask header seq: 7894
272
+ stamp:
273
+ secs: 1733881935
274
+ nsecs: 402739286
275
+ frame_id: "camera_color_optical_frame"
276
+ [INFO] [1733881935.837922]: received mask_obs with header seq: 27
277
+ stamp:
278
+ secs: 1733881935
279
+ nsecs: 837003231
280
+ frame_id: '' and grasped_object_mask header seq: 7900
281
+ stamp:
282
+ secs: 1733881935
283
+ nsecs: 602580309
284
+ frame_id: "camera_color_optical_frame"
285
+ [INFO] [1733881936.034971]: received mask_obs with header seq: 28
286
+ stamp:
287
+ secs: 1733881936
288
+ nsecs: 33885240
289
+ frame_id: '' and grasped_object_mask header seq: 7906
290
+ stamp:
291
+ secs: 1733881935
292
+ nsecs: 802418470
293
+ frame_id: "camera_color_optical_frame"
294
+ [INFO] [1733881936.223721]: received mask_obs with header seq: 29
295
+ stamp:
296
+ secs: 1733881936
297
+ nsecs: 222499608
298
+ frame_id: '' and grasped_object_mask header seq: 7912
299
+ stamp:
300
+ secs: 1733881936
301
+ nsecs: 2255917
302
+ frame_id: "camera_color_optical_frame"
303
+ [INFO] [1733881936.446193]: received mask_obs with header seq: 30
304
+ stamp:
305
+ secs: 1733881936
306
+ nsecs: 445185184
307
+ frame_id: '' and grasped_object_mask header seq: 7918
308
+ stamp:
309
+ secs: 1733881936
310
+ nsecs: 202094316
311
+ frame_id: "camera_color_optical_frame"
312
+ [INFO] [1733881936.632934]: received mask_obs with header seq: 31
313
+ stamp:
314
+ secs: 1733881936
315
+ nsecs: 631910800
316
+ frame_id: '' and grasped_object_mask header seq: 7924
317
+ stamp:
318
+ secs: 1733881936
319
+ nsecs: 401931524
320
+ frame_id: "camera_color_optical_frame"
321
+ [INFO] [1733881936.838232]: received mask_obs with header seq: 32
322
+ stamp:
323
+ secs: 1733881936
324
+ nsecs: 837055444
325
+ frame_id: '' and grasped_object_mask header seq: 7930
326
+ stamp:
327
+ secs: 1733881936
328
+ nsecs: 601768732
329
+ frame_id: "camera_color_optical_frame"
330
+ [INFO] [1733881937.032954]: received mask_obs with header seq: 33
331
+ stamp:
332
+ secs: 1733881937
333
+ nsecs: 32030105
334
+ frame_id: '' and grasped_object_mask header seq: 7936
335
+ stamp:
336
+ secs: 1733881936
337
+ nsecs: 801601171
338
+ frame_id: "camera_color_optical_frame"
339
+ [INFO] [1733881937.227681]: received mask_obs with header seq: 34
340
+ stamp:
341
+ secs: 1733881937
342
+ nsecs: 225735902
343
+ frame_id: '' and grasped_object_mask header seq: 7942
344
+ stamp:
345
+ secs: 1733881937
346
+ nsecs: 1431465
347
+ frame_id: "camera_color_optical_frame"
348
+ [INFO] [1733881937.424884]: received mask_obs with header seq: 35
349
+ stamp:
350
+ secs: 1733881937
351
+ nsecs: 423624753
352
+ frame_id: '' and grasped_object_mask header seq: 7948
353
+ stamp:
354
+ secs: 1733881937
355
+ nsecs: 201261759
356
+ frame_id: "camera_color_optical_frame"
357
+ [INFO] [1733881937.602132]: received mask_obs with header seq: 36
358
+ stamp:
359
+ secs: 1733881937
360
+ nsecs: 601197004
361
+ frame_id: '' and grasped_object_mask header seq: 7953
362
+ stamp:
363
+ secs: 1733881937
364
+ nsecs: 367787123
365
+ frame_id: "camera_color_optical_frame"
366
+ [INFO] [1733881937.819760]: received mask_obs with header seq: 37
367
+ stamp:
368
+ secs: 1733881937
369
+ nsecs: 817212343
370
+ frame_id: '' and grasped_object_mask header seq: 7959
371
+ stamp:
372
+ secs: 1733881937
373
+ nsecs: 567616940
374
+ frame_id: "camera_color_optical_frame"
375
+ [INFO] [1733881938.023777]: received mask_obs with header seq: 38
376
+ stamp:
377
+ secs: 1733881938
378
+ nsecs: 22439956
379
+ frame_id: '' and grasped_object_mask header seq: 7965
380
+ stamp:
381
+ secs: 1733881937
382
+ nsecs: 767450571
383
+ frame_id: "camera_color_optical_frame"
384
+ [INFO] [1733881938.203714]: received mask_obs with header seq: 39
385
+ stamp:
386
+ secs: 1733881938
387
+ nsecs: 202681779
388
+ frame_id: '' and grasped_object_mask header seq: 7971
389
+ stamp:
390
+ secs: 1733881937
391
+ nsecs: 967287064
392
+ frame_id: "camera_color_optical_frame"
393
+ [INFO] [1733881938.403625]: received mask_obs with header seq: 40
394
+ stamp:
395
+ secs: 1733881938
396
+ nsecs: 402268409
397
+ frame_id: '' and grasped_object_mask header seq: 7977
398
+ stamp:
399
+ secs: 1733881938
400
+ nsecs: 167124510
401
+ frame_id: "camera_color_optical_frame"
402
+ [INFO] [1733881938.600104]: received mask_obs with header seq: 41
403
+ stamp:
404
+ secs: 1733881938
405
+ nsecs: 598839759
406
+ frame_id: '' and grasped_object_mask header seq: 7983
407
+ stamp:
408
+ secs: 1733881938
409
+ nsecs: 366960764
410
+ frame_id: "camera_color_optical_frame"
411
+ [INFO] [1733881938.820781]: received mask_obs with header seq: 42
412
+ stamp:
413
+ secs: 1733881938
414
+ nsecs: 819800138
415
+ frame_id: '' and grasped_object_mask header seq: 7989
416
+ stamp:
417
+ secs: 1733881938
418
+ nsecs: 566796780
419
+ frame_id: "camera_color_optical_frame"
420
+ [INFO] [1733881939.014330]: received mask_obs with header seq: 43
421
+ stamp:
422
+ secs: 1733881939
423
+ nsecs: 13423442
424
+ frame_id: '' and grasped_object_mask header seq: 7996
425
+ stamp:
426
+ secs: 1733881938
427
+ nsecs: 799937010
428
+ frame_id: "camera_color_optical_frame"
429
+ [INFO] [1733881939.224791]: received mask_obs with header seq: 44
430
+ stamp:
431
+ secs: 1733881939
432
+ nsecs: 223258733
433
+ frame_id: '' and grasped_object_mask header seq: 8001
434
+ stamp:
435
+ secs: 1733881938
436
+ nsecs: 966463566
437
+ frame_id: "camera_color_optical_frame"
438
+ [INFO] [1733881939.425529]: received mask_obs with header seq: 45
439
+ stamp:
440
+ secs: 1733881939
441
+ nsecs: 424565792
442
+ frame_id: '' and grasped_object_mask header seq: 8008
443
+ stamp:
444
+ secs: 1733881939
445
+ nsecs: 199602365
446
+ frame_id: "camera_color_optical_frame"
447
+ [INFO] [1733881939.619431]: received mask_obs with header seq: 46
448
+ stamp:
449
+ secs: 1733881939
450
+ nsecs: 618312597
451
+ frame_id: '' and grasped_object_mask header seq: 8014
452
+ stamp:
453
+ secs: 1733881939
454
+ nsecs: 399434566
455
+ frame_id: "camera_color_optical_frame"
456
+ [INFO] [1733881939.820555]: received mask_obs with header seq: 47
457
+ stamp:
458
+ secs: 1733881939
459
+ nsecs: 819616794
460
+ frame_id: '' and grasped_object_mask header seq: 8020
461
+ stamp:
462
+ secs: 1733881939
463
+ nsecs: 599266529
464
+ frame_id: "camera_color_optical_frame"
465
+ [INFO] [1733881940.005593]: received mask_obs with header seq: 48
466
+ stamp:
467
+ secs: 1733881940
468
+ nsecs: 4656076
469
+ frame_id: '' and grasped_object_mask header seq: 8025
470
+ stamp:
471
+ secs: 1733881939
472
+ nsecs: 765801430
473
+ frame_id: "camera_color_optical_frame"
474
+ [INFO] [1733881940.204921]: received mask_obs with header seq: 49
475
+ stamp:
476
+ secs: 1733881940
477
+ nsecs: 203764915
478
+ frame_id: '' and grasped_object_mask header seq: 8031
479
+ stamp:
480
+ secs: 1733881939
481
+ nsecs: 965646029
482
+ frame_id: "camera_color_optical_frame"
483
+ [INFO] [1733881940.413592]: received mask_obs with header seq: 50
484
+ stamp:
485
+ secs: 1733881940
486
+ nsecs: 412583351
487
+ frame_id: '' and grasped_object_mask header seq: 8037
488
+ stamp:
489
+ secs: 1733881940
490
+ nsecs: 165490627
491
+ frame_id: "camera_color_optical_frame"
492
+ [INFO] [1733881940.618829]: received mask_obs with header seq: 51
493
+ stamp:
494
+ secs: 1733881940
495
+ nsecs: 617256879
496
+ frame_id: '' and grasped_object_mask header seq: 8043
497
+ stamp:
498
+ secs: 1733881940
499
+ nsecs: 365335941
500
+ frame_id: "camera_color_optical_frame"
501
+ [INFO] [1733881940.824649]: received mask_obs with header seq: 52
502
+ stamp:
503
+ secs: 1733881940
504
+ nsecs: 823161363
505
+ frame_id: '' and grasped_object_mask header seq: 8050
506
+ stamp:
507
+ secs: 1733881940
508
+ nsecs: 598489761
509
+ frame_id: "camera_color_optical_frame"
510
+ [INFO] [1733881941.027861]: received mask_obs with header seq: 53
511
+ stamp:
512
+ secs: 1733881941
513
+ nsecs: 26394367
514
+ frame_id: '' and grasped_object_mask header seq: 8056
515
+ stamp:
516
+ secs: 1733881940
517
+ nsecs: 798328876
518
+ frame_id: "camera_color_optical_frame"
519
+ [INFO] [1733881941.225607]: received mask_obs with header seq: 54
520
+ stamp:
521
+ secs: 1733881941
522
+ nsecs: 224196434
523
+ frame_id: '' and grasped_object_mask header seq: 8061
524
+ stamp:
525
+ secs: 1733881940
526
+ nsecs: 964860439
527
+ frame_id: "camera_color_optical_frame"
528
+ [INFO] [1733881941.403943]: received mask_obs with header seq: 55
529
+ stamp:
530
+ secs: 1733881941
531
+ nsecs: 402922868
532
+ frame_id: '' and grasped_object_mask header seq: 8067
533
+ stamp:
534
+ secs: 1733881941
535
+ nsecs: 164697409
536
+ frame_id: "camera_color_optical_frame"
537
+ [INFO] [1733881941.609296]: received mask_obs with header seq: 56
538
+ stamp:
539
+ secs: 1733881941
540
+ nsecs: 608402252
541
+ frame_id: '' and grasped_object_mask header seq: 8073
542
+ stamp:
543
+ secs: 1733881941
544
+ nsecs: 364534378
545
+ frame_id: "camera_color_optical_frame"
546
+ [INFO] [1733881941.801054]: received mask_obs with header seq: 57
547
+ stamp:
548
+ secs: 1733881941
549
+ nsecs: 799741744
550
+ frame_id: '' and grasped_object_mask header seq: 8079
551
+ stamp:
552
+ secs: 1733881941
553
+ nsecs: 564371347
554
+ frame_id: "camera_color_optical_frame"
555
+ [INFO] [1733881942.001835]: received mask_obs with header seq: 58
556
+ stamp:
557
+ secs: 1733881942
558
+ nsecs: 708103
559
+ frame_id: '' and grasped_object_mask header seq: 8085
560
+ stamp:
561
+ secs: 1733881941
562
+ nsecs: 764206648
563
+ frame_id: "camera_color_optical_frame"
564
+ [INFO] [1733881942.199729]: received mask_obs with header seq: 59
565
+ stamp:
566
+ secs: 1733881942
567
+ nsecs: 198785305
568
+ frame_id: '' and grasped_object_mask header seq: 8091
569
+ stamp:
570
+ secs: 1733881941
571
+ nsecs: 964039087
572
+ frame_id: "camera_color_optical_frame"
573
+ [INFO] [1733881942.426365]: received mask_obs with header seq: 60
574
+ stamp:
575
+ secs: 1733881942
576
+ nsecs: 425239324
577
+ frame_id: '' and grasped_object_mask header seq: 8097
578
+ stamp:
579
+ secs: 1733881942
580
+ nsecs: 163871288
581
+ frame_id: "camera_color_optical_frame"
582
+ [INFO] [1733881942.619282]: received mask_obs with header seq: 61
583
+ stamp:
584
+ secs: 1733881942
585
+ nsecs: 617688417
586
+ frame_id: '' and grasped_object_mask header seq: 8104
587
+ stamp:
588
+ secs: 1733881942
589
+ nsecs: 397009134
590
+ frame_id: "camera_color_optical_frame"
591
+ [INFO] [1733881942.806585]: received mask_obs with header seq: 62
592
+ stamp:
593
+ secs: 1733881942
594
+ nsecs: 805705308
595
+ frame_id: '' and grasped_object_mask header seq: 8109
596
+ stamp:
597
+ secs: 1733881942
598
+ nsecs: 563535213
599
+ frame_id: "camera_color_optical_frame"
600
+ [INFO] [1733881943.003788]: received mask_obs with header seq: 63
601
+ stamp:
602
+ secs: 1733881943
603
+ nsecs: 2786159
604
+ frame_id: '' and grasped_object_mask header seq: 8115
605
+ stamp:
606
+ secs: 1733881942
607
+ nsecs: 763370991
608
+ frame_id: "camera_color_optical_frame"
609
+ [INFO] [1733881943.203472]: received mask_obs with header seq: 64
610
+ stamp:
611
+ secs: 1733881943
612
+ nsecs: 202393293
613
+ frame_id: '' and grasped_object_mask header seq: 8121
614
+ stamp:
615
+ secs: 1733881942
616
+ nsecs: 963209629
617
+ frame_id: "camera_color_optical_frame"
618
+ [INFO] [1733881943.406879]: received mask_obs with header seq: 65
619
+ stamp:
620
+ secs: 1733881943
621
+ nsecs: 405942440
622
+ frame_id: '' and grasped_object_mask header seq: 8127
623
+ stamp:
624
+ secs: 1733881943
625
+ nsecs: 163049221
626
+ frame_id: "camera_color_optical_frame"
627
+ [INFO] [1733881943.602010]: received mask_obs with header seq: 66
628
+ stamp:
629
+ secs: 1733881943
630
+ nsecs: 600990533
631
+ frame_id: '' and grasped_object_mask header seq: 8133
632
+ stamp:
633
+ secs: 1733881943
634
+ nsecs: 362887859
635
+ frame_id: "camera_color_optical_frame"
636
+ [INFO] [1733881943.805436]: received mask_obs with header seq: 67
637
+ stamp:
638
+ secs: 1733881943
639
+ nsecs: 804544448
640
+ frame_id: '' and grasped_object_mask header seq: 8139
641
+ stamp:
642
+ secs: 1733881943
643
+ nsecs: 562726736
644
+ frame_id: "camera_color_optical_frame"
645
+ [INFO] [1733881944.003641]: received mask_obs with header seq: 68
646
+ stamp:
647
+ secs: 1733881944
648
+ nsecs: 2423763
649
+ frame_id: '' and grasped_object_mask header seq: 8145
650
+ stamp:
651
+ secs: 1733881943
652
+ nsecs: 762568951
653
+ frame_id: "camera_color_optical_frame"
654
+ [INFO] [1733881944.201632]: received mask_obs with header seq: 69
655
+ stamp:
656
+ secs: 1733881944
657
+ nsecs: 200727224
658
+ frame_id: '' and grasped_object_mask header seq: 8151
659
+ stamp:
660
+ secs: 1733881943
661
+ nsecs: 962414980
662
+ frame_id: "camera_color_optical_frame"
663
+ [INFO] [1733881944.411565]: received mask_obs with header seq: 70
664
+ stamp:
665
+ secs: 1733881944
666
+ nsecs: 410659313
667
+ frame_id: '' and grasped_object_mask header seq: 8157
668
+ stamp:
669
+ secs: 1733881944
670
+ nsecs: 162261248
671
+ frame_id: "camera_color_optical_frame"
672
+ [INFO] [1733881944.608529]: received mask_obs with header seq: 71
673
+ stamp:
674
+ secs: 1733881944
675
+ nsecs: 607629060
676
+ frame_id: '' and grasped_object_mask header seq: 8163
677
+ stamp:
678
+ secs: 1733881944
679
+ nsecs: 362109184
680
+ frame_id: "camera_color_optical_frame"
681
+ [INFO] [1733881944.799691]: received mask_obs with header seq: 72
682
+ stamp:
683
+ secs: 1733881944
684
+ nsecs: 798696279
685
+ frame_id: '' and grasped_object_mask header seq: 8169
686
+ stamp:
687
+ secs: 1733881944
688
+ nsecs: 561956167
689
+ frame_id: "camera_color_optical_frame"
690
+ [INFO] [1733881945.002218]: received mask_obs with header seq: 73
691
+ stamp:
692
+ secs: 1733881945
693
+ nsecs: 1061439
694
+ frame_id: '' and grasped_object_mask header seq: 8175
695
+ stamp:
696
+ secs: 1733881944
697
+ nsecs: 761801004
698
+ frame_id: "camera_color_optical_frame"
699
+ [INFO] [1733881945.202708]: received mask_obs with header seq: 74
700
+ stamp:
701
+ secs: 1733881945
702
+ nsecs: 201566219
703
+ frame_id: '' and grasped_object_mask header seq: 8181
704
+ stamp:
705
+ secs: 1733881944
706
+ nsecs: 961643934
707
+ frame_id: "camera_color_optical_frame"
708
+ [INFO] [1733881945.408593]: received mask_obs with header seq: 75
709
+ stamp:
710
+ secs: 1733881945
711
+ nsecs: 407604694
712
+ frame_id: '' and grasped_object_mask header seq: 8187
713
+ stamp:
714
+ secs: 1733881945
715
+ nsecs: 161487103
716
+ frame_id: "camera_color_optical_frame"
717
+ [INFO] [1733881945.601663]: received mask_obs with header seq: 76
718
+ stamp:
719
+ secs: 1733881945
720
+ nsecs: 600501298
721
+ frame_id: '' and grasped_object_mask header seq: 8193
722
+ stamp:
723
+ secs: 1733881945
724
+ nsecs: 361330271
725
+ frame_id: "camera_color_optical_frame"
726
+ [INFO] [1733881945.808530]: received mask_obs with header seq: 77
727
+ stamp:
728
+ secs: 1733881945
729
+ nsecs: 807325363
730
+ frame_id: '' and grasped_object_mask header seq: 8199
731
+ stamp:
732
+ secs: 1733881945
733
+ nsecs: 561174870
734
+ frame_id: "camera_color_optical_frame"
735
+ [INFO] [1733881946.008040]: received mask_obs with header seq: 78
736
+ stamp:
737
+ secs: 1733881946
738
+ nsecs: 7133245
739
+ frame_id: '' and grasped_object_mask header seq: 8205
740
+ stamp:
741
+ secs: 1733881945
742
+ nsecs: 761015177
743
+ frame_id: "camera_color_optical_frame"
744
+ [INFO] [1733881946.201357]: received mask_obs with header seq: 79
745
+ stamp:
746
+ secs: 1733881946
747
+ nsecs: 200472593
748
+ frame_id: '' and grasped_object_mask header seq: 8211
749
+ stamp:
750
+ secs: 1733881945
751
+ nsecs: 960851908
752
+ frame_id: "camera_color_optical_frame"
753
+ [INFO] [1733881946.401407]: received mask_obs with header seq: 80
754
+ stamp:
755
+ secs: 1733881946
756
+ nsecs: 400375843
757
+ frame_id: '' and grasped_object_mask header seq: 8217
758
+ stamp:
759
+ secs: 1733881946
760
+ nsecs: 160688162
761
+ frame_id: "camera_color_optical_frame"
762
+ [INFO] [1733881946.607615]: received mask_obs with header seq: 81
763
+ stamp:
764
+ secs: 1733881946
765
+ nsecs: 606504678
766
+ frame_id: '' and grasped_object_mask header seq: 8223
767
+ stamp:
768
+ secs: 1733881946
769
+ nsecs: 360524893
770
+ frame_id: "camera_color_optical_frame"
771
+ [INFO] [1733881946.810744]: received mask_obs with header seq: 82
772
+ stamp:
773
+ secs: 1733881946
774
+ nsecs: 809859037
775
+ frame_id: '' and grasped_object_mask header seq: 8229
776
+ stamp:
777
+ secs: 1733881946
778
+ nsecs: 560361385
779
+ frame_id: "camera_color_optical_frame"
780
+ [INFO] [1733881947.001563]: received mask_obs with header seq: 83
781
+ stamp:
782
+ secs: 1733881947
783
+ nsecs: 424861
784
+ frame_id: '' and grasped_object_mask header seq: 8235
785
+ stamp:
786
+ secs: 1733881946
787
+ nsecs: 760201216
788
+ frame_id: "camera_color_optical_frame"
789
+ [INFO] [1733881947.203573]: received mask_obs with header seq: 84
790
+ stamp:
791
+ secs: 1733881947
792
+ nsecs: 202486991
793
+ frame_id: '' and grasped_object_mask header seq: 8241
794
+ stamp:
795
+ secs: 1733881946
796
+ nsecs: 960042477
797
+ frame_id: "camera_color_optical_frame"
798
+ [INFO] [1733881947.405680]: received mask_obs with header seq: 85
799
+ stamp:
800
+ secs: 1733881947
801
+ nsecs: 404779672
802
+ frame_id: '' and grasped_object_mask header seq: 8247
803
+ stamp:
804
+ secs: 1733881947
805
+ nsecs: 159883976
806
+ frame_id: "camera_color_optical_frame"
807
+ [INFO] [1733881947.599595]: received mask_obs with header seq: 86
808
+ stamp:
809
+ secs: 1733881947
810
+ nsecs: 598672866
811
+ frame_id: '' and grasped_object_mask header seq: 8253
812
+ stamp:
813
+ secs: 1733881947
814
+ nsecs: 359725475
815
+ frame_id: "camera_color_optical_frame"
816
+ [INFO] [1733881947.800599]: received mask_obs with header seq: 87
817
+ stamp:
818
+ secs: 1733881947
819
+ nsecs: 799697399
820
+ frame_id: '' and grasped_object_mask header seq: 8259
821
+ stamp:
822
+ secs: 1733881947
823
+ nsecs: 559567451
824
+ frame_id: "camera_color_optical_frame"
825
+ [INFO] [1733881947.999780]: received mask_obs with header seq: 88
826
+ stamp:
827
+ secs: 1733881947
828
+ nsecs: 998712778
829
+ frame_id: '' and grasped_object_mask header seq: 8265
830
+ stamp:
831
+ secs: 1733881947
832
+ nsecs: 759409428
833
+ frame_id: "camera_color_optical_frame"
834
+ [INFO] [1733881948.207681]: received mask_obs with header seq: 89
835
+ stamp:
836
+ secs: 1733881948
837
+ nsecs: 206808567
838
+ frame_id: '' and grasped_object_mask header seq: 8271
839
+ stamp:
840
+ secs: 1733881947
841
+ nsecs: 959252834
842
+ frame_id: "camera_color_optical_frame"
843
+ [INFO] [1733881948.404581]: received mask_obs with header seq: 90
844
+ stamp:
845
+ secs: 1733881948
846
+ nsecs: 403736352
847
+ frame_id: '' and grasped_object_mask header seq: 8277
848
+ stamp:
849
+ secs: 1733881948
850
+ nsecs: 159095764
851
+ frame_id: "camera_color_optical_frame"
852
+ [INFO] [1733881948.607006]: received mask_obs with header seq: 91
853
+ stamp:
854
+ secs: 1733881948
855
+ nsecs: 605962514
856
+ frame_id: '' and grasped_object_mask header seq: 8283
857
+ stamp:
858
+ secs: 1733881948
859
+ nsecs: 358938456
860
+ frame_id: "camera_color_optical_frame"
861
+ [INFO] [1733881948.826154]: received mask_obs with header seq: 92
862
+ stamp:
863
+ secs: 1733881948
864
+ nsecs: 824640035
865
+ frame_id: '' and grasped_object_mask header seq: 8289
866
+ stamp:
867
+ secs: 1733881948
868
+ nsecs: 558781624
869
+ frame_id: "camera_color_optical_frame"
870
+ [INFO] [1733881949.042130]: received mask_obs with header seq: 93
871
+ stamp:
872
+ secs: 1733881949
873
+ nsecs: 41005611
874
+ frame_id: '' and grasped_object_mask header seq: 8296
875
+ stamp:
876
+ secs: 1733881948
877
+ nsecs: 791931868
878
+ frame_id: "camera_color_optical_frame"
879
+ [INFO] [1733881949.253016]: received mask_obs with header seq: 94
880
+ stamp:
881
+ secs: 1733881949
882
+ nsecs: 251410484
883
+ frame_id: '' and grasped_object_mask header seq: 8302
884
+ stamp:
885
+ secs: 1733881948
886
+ nsecs: 991774082
887
+ frame_id: "camera_color_optical_frame"
888
+ [INFO] [1733881949.451710]: received mask_obs with header seq: 95
889
+ stamp:
890
+ secs: 1733881949
891
+ nsecs: 449931383
892
+ frame_id: '' and grasped_object_mask header seq: 8308
893
+ stamp:
894
+ secs: 1733881949
895
+ nsecs: 191616774
896
+ frame_id: "camera_color_optical_frame"
897
+ For topic gripper_width: timestamp difference is 110894429 for nearest: 1733881949502353656 - target: 1733881949391459227 at idx 249
898
+ [INFO] [1733881949.644919]: received mask_obs with header seq: 96
899
+ stamp:
900
+ secs: 1733881949
901
+ nsecs: 644002914
902
+ frame_id: '' and grasped_object_mask header seq: 8314
903
+ stamp:
904
+ secs: 1733881949
905
+ nsecs: 391459227
906
+ frame_id: "camera_color_optical_frame"
907
+ [INFO] [1733881949.839658]: received mask_obs with header seq: 97
908
+ stamp:
909
+ secs: 1733881949
910
+ nsecs: 838388442
911
+ frame_id: '' and grasped_object_mask header seq: 8320
912
+ stamp:
913
+ secs: 1733881949
914
+ nsecs: 591301680
915
+ frame_id: "camera_color_optical_frame"
916
+ [INFO] [1733881950.026284]: received mask_obs with header seq: 98
917
+ stamp:
918
+ secs: 1733881950
919
+ nsecs: 24919033
920
+ frame_id: '' and grasped_object_mask header seq: 8326
921
+ stamp:
922
+ secs: 1733881949
923
+ nsecs: 791146994
924
+ frame_id: "camera_color_optical_frame"
925
+ [INFO] [1733881950.241674]: received mask_obs with header seq: 99
926
+ stamp:
927
+ secs: 1733881950
928
+ nsecs: 240784645
929
+ frame_id: '' and grasped_object_mask header seq: 8332
930
+ stamp:
931
+ secs: 1733881949
932
+ nsecs: 990992785
933
+ frame_id: "camera_color_optical_frame"
934
+ [INFO] [1733881950.441060]: received mask_obs with header seq: 100
935
+ stamp:
936
+ secs: 1733881950
937
+ nsecs: 439967393
938
+ frame_id: '' and grasped_object_mask header seq: 8338
939
+ stamp:
940
+ secs: 1733881950
941
+ nsecs: 190838814
942
+ frame_id: "camera_color_optical_frame"
943
+ [INFO] [1733881950.630870]: received mask_obs with header seq: 101
944
+ stamp:
945
+ secs: 1733881950
946
+ nsecs: 629913806
947
+ frame_id: '' and grasped_object_mask header seq: 8344
948
+ stamp:
949
+ secs: 1733881950
950
+ nsecs: 390685320
951
+ frame_id: "camera_color_optical_frame"
952
+ [INFO] [1733881950.832320]: received mask_obs with header seq: 102
953
+ stamp:
954
+ secs: 1733881950
955
+ nsecs: 831316471
956
+ frame_id: '' and grasped_object_mask header seq: 8350
957
+ stamp:
958
+ secs: 1733881950
959
+ nsecs: 590532303
960
+ frame_id: "camera_color_optical_frame"
961
+ [INFO] [1733881951.028132]: received mask_obs with header seq: 103
962
+ stamp:
963
+ secs: 1733881951
964
+ nsecs: 27282714
965
+ frame_id: '' and grasped_object_mask header seq: 8356
966
+ stamp:
967
+ secs: 1733881950
968
+ nsecs: 790374041
969
+ frame_id: "camera_color_optical_frame"
970
+ [INFO] [1733881951.215708]: received mask_obs with header seq: 104
971
+ stamp:
972
+ secs: 1733881951
973
+ nsecs: 214742898
974
+ frame_id: '' and grasped_object_mask header seq: 8362
975
+ stamp:
976
+ secs: 1733881950
977
+ nsecs: 990212679
978
+ frame_id: "camera_color_optical_frame"
979
+ [INFO] [1733881951.419554]: received mask_obs with header seq: 105
980
+ stamp:
981
+ secs: 1733881951
982
+ nsecs: 418620824
983
+ frame_id: '' and grasped_object_mask header seq: 8368
984
+ stamp:
985
+ secs: 1733881951
986
+ nsecs: 190052032
987
+ frame_id: "camera_color_optical_frame"
988
+ [rosbag_record-3] killing on exit
989
+ [pub_text-1] killing on exit[print_text-2] killing on exit
990
+ [INFO] [1733881951.619548]: received mask_obs with header seq: 106
991
+ stamp:
992
+ secs: 1733881951
993
+ nsecs: 603054285
994
+ frame_id: '' and grasped_object_mask header seq: 8373
995
+ stamp:
996
+ secs: 1733881951
997
+ nsecs: 356583834
998
+ frame_id: "camera_color_optical_frame"
999
+ [INFO] [1733881951.832582]: received mask_obs with header seq: 107
1000
+ stamp:
1001
+ secs: 1733881951
1002
+ nsecs: 831149339
1003
+ frame_id: '' and grasped_object_mask header seq: 8380
1004
+ stamp:
1005
+ secs: 1733881951
1006
+ nsecs: 589729309
1007
+ frame_id: "camera_color_optical_frame"
1008
+ [INFO] [1733881952.050082]: received mask_obs with header seq: 108
1009
+ stamp:
1010
+ secs: 1733881952
1011
+ nsecs: 46733140
1012
+ frame_id: '' and grasped_object_mask header seq: 8386
1013
+ stamp:
1014
+ secs: 1733881951
1015
+ nsecs: 789563179
1016
+ frame_id: "camera_color_optical_frame"
1017
+ [INFO] [1733881952.244104]: received mask_obs with header seq: 109
1018
+ stamp:
1019
+ secs: 1733881952
1020
+ nsecs: 243089437
1021
+ frame_id: '' and grasped_object_mask header seq: 8393
1022
+ stamp:
1023
+ secs: 1733881952
1024
+ nsecs: 22698879
1025
+ frame_id: "camera_color_optical_frame"
1026
+ Exception in thread Thread-62:
1027
+ Traceback (most recent call last):
1028
+ File "/home/leonmkim/.pyenv/versions/3.10.14/lib/python3.10/threading.py", line 1016, in _bootstrap_inner
1029
+ self.run()
1030
+ File "/opt/ros/noetic/lib/python3/dist-packages/rospy/timer.py", line 237, in run
1031
+ self._callback(TimerEvent(last_expected, last_real, current_expected, current_real, last_duration))
1032
+ File "/home/leonmkim/fish_leon/FISH/video.py", line 78, in ros_record_cb
1033
+ self.process.stdin.write(frame.tobytes())
1034
+ BrokenPipeError: [Errno 32] Broken pipe
1035
+ program exiting gracefully
1036
+ Exception in thread Thread-71:
1037
+ Traceback (most recent call last):
1038
+ File "/home/leonmkim/.pyenv/versions/3.10.14/lib/python3.10/threading.py", line 1016, in _bootstrap_inner
1039
+ self.run()
1040
+ File "/opt/ros/noetic/lib/python3/dist-packages/rospy/timer.py", line 237, in run
1041
+ self._callback(TimerEvent(last_expected, last_real, current_expected, current_real, last_duration))
1042
+ File "/home/leonmkim/fish_leon/gym-envs/gym_envs/envs/franka_env.py", line 585, in ros_step_cb
1043
+ mask_header, mask_obs_msg = self.wait_for_exact_match(frame_metadata['rgb_header'].stamp, self.mask_obs_queue)
1044
+ File "/home/leonmkim/fish_leon/gym-envs/gym_envs/envs/franka_env.py", line 636, in wait_for_exact_match
1045
+ assert queue_return is not None, f"Exact match not found in {timeout} seconds"
1046
+ AssertionError: Exact match not found in 5 seconds
1047
+ proceeding to start episode!
1048
+ [INFO] [1733881964.331730]: Saving video...
1049
+ [INFO] [1733881964.332162]: Video saved!
205123/wandb/run-20241210_205130-svysfc56/files/requirements.txt ADDED
@@ -0,0 +1,339 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ Cython==3.0.10
2
+ Farama-Notifications==0.0.4
3
+ GitPython==3.1.43
4
+ Jinja2==3.1.4
5
+ Markdown==3.6
6
+ MarkupSafe==2.1.5
7
+ POT==0.7.0
8
+ PyOpenGL==3.1.7
9
+ PySocks==1.7.1
10
+ PyYAML==6.0.1
11
+ Pygments==2.18.0
12
+ Rtree==1.3.0
13
+ Werkzeug==3.0.3
14
+ absl-py==2.1.0
15
+ accelerate==0.33.0
16
+ actionlib-msgs==1.13.0.post3
17
+ actionlib==1.12.0
18
+ actionlib==1.14.0
19
+ aiohttp==3.9.5
20
+ aiosignal==1.3.1
21
+ angles==1.9.13
22
+ antlr4-python3-runtime==4.9.3
23
+ anyio==4.4.0
24
+ asciitree==0.3.3
25
+ async-timeout==4.0.3
26
+ attrs==23.2.0
27
+ autoprop==4.1.0
28
+ beartype==0.18.5
29
+ beautifulsoup4==4.12.3
30
+ bondpy==1.8.6
31
+ byol-pytorch==0.8.0
32
+ cachetools==5.4.0
33
+ camera-calibration-parsers==1.12.0
34
+ camera-calibration==1.17.0
35
+ cascadio==0.0.13
36
+ catkin-pkg==1.0.0
37
+ catkin==0.7.18
38
+ catkin==0.8.10
39
+ certifi==2024.7.4
40
+ cffi==1.16.0
41
+ chardet==5.2.0
42
+ charset-normalizer==3.3.2
43
+ click==8.1.7
44
+ cloudpickle==3.0.0
45
+ cmake==3.30.1
46
+ colorlog==6.8.2
47
+ contourpy==1.2.1
48
+ controller-manager-msgs==0.20.0
49
+ controller-manager==0.20.0
50
+ coverage==7.6.0
51
+ coveralls==4.0.1
52
+ cv-bridge==1.16.2
53
+ cycler==0.12.1
54
+ datasets==2.20.0
55
+ decorator==4.4.2
56
+ deepdiff==7.0.1
57
+ defusedxml==0.7.1
58
+ diagnostic-analysis==1.11.0
59
+ diagnostic-common-diagnostics==1.11.0
60
+ diagnostic-updater==1.11.0
61
+ diffusers==0.27.2
62
+ dill==0.3.8
63
+ distro==1.9.0
64
+ dm-control==1.0.8
65
+ dm-env==1.6
66
+ dm-tree==0.1.8
67
+ docker-pycreds==0.4.0
68
+ docopt==0.6.2
69
+ docutils==0.21.2
70
+ dynamic-reconfigure==1.7.3
71
+ einops==0.8.0
72
+ embreex==2.17.7.post5
73
+ empy==3.3.4
74
+ etils==1.7.0
75
+ exceptiongroup==1.2.2
76
+ ezdxf==1.3.2
77
+ fasteners==0.19
78
+ filelock==3.15.4
79
+ fonttools==4.53.1
80
+ freetype-py==2.4.0
81
+ frozenlist==1.4.1
82
+ fsspec==2024.5.0
83
+ gazebo_plugins==2.9.2
84
+ gazebo_ros==2.9.2
85
+ gdown==5.2.0
86
+ gencpp==0.7.0
87
+ geneus==3.0.0
88
+ genlisp==0.4.18
89
+ genmsg==0.5.12
90
+ genmsg==0.6.0
91
+ gennodejs==2.0.2
92
+ genpy==0.6.14
93
+ genpy==0.6.15
94
+ geometry-msgs==1.13.0.post2
95
+ gitdb==4.0.11
96
+ glfw==2.7.0
97
+ glooey==0.3.6
98
+ gmsh==4.12.2
99
+ gnupg==2.3.1
100
+ google-auth-oauthlib==1.0.0
101
+ google-auth==2.32.0
102
+ grpcio==1.65.1
103
+ gym-envs==0.0.1
104
+ gym-notices==0.0.8
105
+ gym==0.22.0
106
+ gymnasium==0.29.1
107
+ h11==0.14.0
108
+ h5py==3.11.0
109
+ hf_transfer==0.1.8
110
+ httpcore==1.0.5
111
+ httpx==0.27.0
112
+ huggingface-hub==0.23.5
113
+ hydra-core==1.3.2
114
+ hydra-submitit-launcher==1.2.0
115
+ idna==3.7
116
+ image-geometry==1.16.2
117
+ imageio-ffmpeg==0.5.1
118
+ imageio==2.34.2
119
+ importlib_metadata==8.2.0
120
+ importlib_resources==6.4.0
121
+ iniconfig==2.0.0
122
+ interactive-markers==1.12.0
123
+ joint-state-publisher-gui==1.15.1
124
+ joint-state-publisher==1.15.1
125
+ jsonschema-specifications==2023.12.1
126
+ jsonschema==4.23.0
127
+ kiwisolver==1.4.5
128
+ kornia==0.7.3
129
+ kornia_rs==0.1.5
130
+ labmaze==1.0.6
131
+ laser_geometry==1.6.7
132
+ lazy_loader==0.4
133
+ lerobot==0.1.0
134
+ lightning-utilities==0.11.6
135
+ llvmlite==0.43.0
136
+ lxml==5.2.2
137
+ manifold3d==2.5.1
138
+ mapbox-earcut==1.0.1
139
+ markdown-it-py==3.0.0
140
+ matplotlib==3.9.1
141
+ mdurl==0.1.2
142
+ meshio==5.3.5
143
+ message-filters==1.16.0
144
+ more-itertools==10.3.0
145
+ moviepy==1.0.3
146
+ mpmath==1.3.0
147
+ mujoco==3.2.0
148
+ multidict==6.0.5
149
+ multiprocess==0.70.16
150
+ natsort==8.4.0
151
+ netifaces==0.11.0
152
+ networkx==3.3
153
+ nodeenv==1.9.1
154
+ numba==0.60.0
155
+ numcodecs==0.13.0
156
+ numpy==1.26.4
157
+ nvidia-cublas-cu12==12.1.3.1
158
+ nvidia-cuda-cupti-cu12==12.1.105
159
+ nvidia-cuda-nvrtc-cu12==12.1.105
160
+ nvidia-cuda-runtime-cu12==12.1.105
161
+ nvidia-cudnn-cu12==9.1.0.70
162
+ nvidia-cufft-cu12==11.0.2.54
163
+ nvidia-curand-cu12==10.3.2.106
164
+ nvidia-cusolver-cu12==11.4.5.107
165
+ nvidia-cusparse-cu12==12.1.0.106
166
+ nvidia-nccl-cu12==2.20.5
167
+ nvidia-nvjitlink-cu12==12.5.82
168
+ nvidia-nvtx-cu12==12.1.105
169
+ oauthlib==3.2.2
170
+ omegaconf==2.3.0
171
+ openctm==0.0.5
172
+ opencv-python==4.10.0.84
173
+ ordered-set==4.1.0
174
+ packaging==24.1
175
+ pandas==2.2.2
176
+ pillow==10.4.0
177
+ pip==24.2
178
+ platformdirs==4.2.2
179
+ pluggy==1.5.0
180
+ proglog==0.1.10
181
+ protobuf==5.27.2
182
+ psutil==6.0.0
183
+ pyarrow-hotfix==0.6
184
+ pyarrow==17.0.0
185
+ pyasn1==0.6.0
186
+ pyasn1_modules==0.4.0
187
+ pyav==12.3.0
188
+ pycollada==0.8
189
+ pycparser==2.22
190
+ pycryptodomex==3.21.0
191
+ pyglet==1.5.29
192
+ pyinstrument==4.6.2
193
+ pymunk==6.8.1
194
+ pyparsing==2.4.7
195
+ pyrealsense2==2.54.2.5684
196
+ pyribbit==0.1.46
197
+ pyright==1.1.373
198
+ pytest-beartype==0.0.2
199
+ pytest-cov==5.0.0
200
+ pytest==8.3.1
201
+ python-dateutil==2.9.0.post0
202
+ python-fcl==0.7.0.6
203
+ python-qt-binding==0.4.4
204
+ pytorch-lightning==2.4.0
205
+ pytz==2024.1
206
+ qt-dotgraph==0.4.2
207
+ qt-gui-cpp==0.4.2
208
+ qt-gui-py-common==0.4.2
209
+ qt-gui==0.4.2
210
+ referencing==0.35.1
211
+ regex==2024.5.15
212
+ requests-oauthlib==2.0.0
213
+ requests==2.32.3
214
+ rerun-sdk==0.17.0
215
+ resource_retriever==1.12.7
216
+ rich==13.7.1
217
+ ros-numpy==0.0.5
218
+ rosbag==1.16.0
219
+ rosboost-cfg==1.15.8
220
+ rosclean==1.15.8
221
+ roscpp==1.15.11
222
+ roscreate==1.15.8
223
+ rosgraph-msgs==1.11.3.post2
224
+ rosgraph==1.15.11
225
+ rosgraph==1.16.0
226
+ roslaunch==1.16.0
227
+ roslib==1.14.7.post0
228
+ roslib==1.15.8
229
+ roslint==0.12.0
230
+ roslz4==1.16.0
231
+ rosmake==1.15.8
232
+ rosmaster==1.16.0
233
+ rosmsg==1.16.0
234
+ rosnode==1.16.0
235
+ rosparam==1.16.0
236
+ rospkg==1.5.1
237
+ rospy==1.15.11
238
+ rospy==1.16.0
239
+ rosservice==1.16.0
240
+ rostest==1.16.0
241
+ rostopic==1.16.0
242
+ rosunit==1.15.8
243
+ roswtf==1.16.0
244
+ rpds-py==0.19.1
245
+ rqt-console==0.4.12
246
+ rqt-image-view==0.4.17
247
+ rqt-logger-level==0.4.12
248
+ rqt-moveit==0.5.11
249
+ rqt-reconfigure==0.5.5
250
+ rqt-robot-dashboard==0.5.8
251
+ rqt-robot-monitor==0.5.15
252
+ rqt-runtime-monitor==0.5.10
253
+ rqt-rviz==0.7.0
254
+ rqt-tf-tree==0.6.4
255
+ rqt_action==0.4.9
256
+ rqt_bag==0.5.1
257
+ rqt_bag_plugins==0.5.1
258
+ rqt_dep==0.4.12
259
+ rqt_graph==0.4.14
260
+ rqt_gui==0.5.3
261
+ rqt_gui_py==0.5.3
262
+ rqt_launch==0.4.9
263
+ rqt_msg==0.4.10
264
+ rqt_nav_view==0.5.7
265
+ rqt_plot==0.4.13
266
+ rqt_pose_view==0.5.11
267
+ rqt_publisher==0.4.10
268
+ rqt_py_common==0.5.3
269
+ rqt_py_console==0.4.10
270
+ rqt_robot_steering==0.5.12
271
+ rqt_service_caller==0.4.10
272
+ rqt_shell==0.4.11
273
+ rqt_srv==0.4.9
274
+ rqt_top==0.4.10
275
+ rqt_topic==0.4.13
276
+ rqt_web==0.4.10
277
+ rsa==4.9
278
+ ruff==0.5.4
279
+ rviz==1.14.25
280
+ safetensors==0.4.3
281
+ scikit-image==0.24.0
282
+ scikit-video==1.1.11
283
+ scipy==1.14.0
284
+ seaborn==0.13.2
285
+ sensor-msgs==1.13.1
286
+ sentry-sdk==2.11.0
287
+ setproctitle==1.3.3
288
+ setuptools==65.5.0
289
+ shapely==2.0.5
290
+ signature_dispatch==1.0.1
291
+ six==1.16.0
292
+ smach-ros==2.5.2
293
+ smach==2.5.2
294
+ smclib==1.8.6
295
+ smmap==5.0.1
296
+ sniffio==1.3.1
297
+ soupsieve==2.5
298
+ std-msgs==0.5.13.post0
299
+ submitit==1.5.1
300
+ svg.path==6.3
301
+ sympy==1.13.1
302
+ tensorboard-data-server==0.7.2
303
+ tensorboard==2.14.0
304
+ termcolor==2.4.0
305
+ tf-conversions==1.13.2
306
+ tf2-geometry-msgs==0.7.7
307
+ tf2-kdl==0.7.7
308
+ tf2-msgs==0.7.2.post3
309
+ tf2-py==0.7.7
310
+ tf2-ros==0.6.5
311
+ tf2-ros==0.7.7
312
+ tf2_py==0.6.5.post1
313
+ tf==1.13.2
314
+ tifffile==2024.7.24
315
+ tomli==2.0.1
316
+ topic-tools==1.16.0
317
+ torch==2.4.0
318
+ torchaug==0.5.2
319
+ torchmetrics==1.4.0.post0
320
+ torchvision==0.19.0
321
+ tqdm==4.66.4
322
+ trimesh==4.4.3
323
+ triton==3.0.0
324
+ typeguard==3.0.2
325
+ typing_extensions==4.12.2
326
+ tzdata==2024.1
327
+ urchin==0.0.27
328
+ urllib3==2.2.2
329
+ vecrec==0.3.1
330
+ vhacdx==0.0.8.post1
331
+ wandb==0.17.5
332
+ wheel==0.43.0
333
+ xacro==1.14.18
334
+ xatlas==0.0.9
335
+ xxhash==3.4.1
336
+ yarl==1.9.4
337
+ zarr==2.18.2
338
+ zipp==3.19.2
339
+ zstandard==0.23.0
205123/wandb/run-20241210_205130-svysfc56/files/wandb-metadata.json ADDED
@@ -0,0 +1,92 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "os": "Linux-5.15.0-125-generic-x86_64-with-glibc2.31",
3
+ "python": "3.10.14",
4
+ "heartbeatAt": "2024-12-11T01:51:30.761221",
5
+ "startedAt": "2024-12-11T01:51:30.302896",
6
+ "docker": null,
7
+ "cuda": null,
8
+ "args": [
9
+ "agent=diffusion",
10
+ "suite=frankagym",
11
+ "suite/frankagym_task@_global_=insertion"
12
+ ],
13
+ "state": "running",
14
+ "program": "/home/leonmkim/fish_leon/FISH/eval_robot.py",
15
+ "codePathLocal": null,
16
+ "codePath": "FISH/eval_robot.py",
17
+ "git": {
18
+ "remote": "https://github.com/leonmkim/fish_leon.git",
19
+ "commit": "4f7c0421eb84a7caf6bf5670f68777523362676f"
20
+ },
21
+ "email": "leonmkim@seas.upenn.edu",
22
+ "root": "/home/leonmkim/fish_leon",
23
+ "host": "leonmkim-ROG-Strix-G15CS-G15CS",
24
+ "username": "leonmkim",
25
+ "executable": "/home/leonmkim/.pyenv/versions/lerobot/bin/python",
26
+ "cpu_count": 8,
27
+ "cpu_count_logical": 8,
28
+ "cpu_freq": {
29
+ "current": 4124.7015,
30
+ "min": 800.0,
31
+ "max": 4700.0
32
+ },
33
+ "cpu_freq_per_core": [
34
+ {
35
+ "current": 3000.0,
36
+ "min": 800.0,
37
+ "max": 4700.0
38
+ },
39
+ {
40
+ "current": 4499.948,
41
+ "min": 800.0,
42
+ "max": 4700.0
43
+ },
44
+ {
45
+ "current": 4499.997,
46
+ "min": 800.0,
47
+ "max": 4700.0
48
+ },
49
+ {
50
+ "current": 3000.0,
51
+ "min": 800.0,
52
+ "max": 4700.0
53
+ },
54
+ {
55
+ "current": 3000.0,
56
+ "min": 800.0,
57
+ "max": 4700.0
58
+ },
59
+ {
60
+ "current": 3000.0,
61
+ "min": 800.0,
62
+ "max": 4700.0
63
+ },
64
+ {
65
+ "current": 3000.0,
66
+ "min": 800.0,
67
+ "max": 4700.0
68
+ },
69
+ {
70
+ "current": 4499.999,
71
+ "min": 800.0,
72
+ "max": 4700.0
73
+ }
74
+ ],
75
+ "disk": {
76
+ "/": {
77
+ "total": 915.3232879638672,
78
+ "used": 364.1780319213867
79
+ }
80
+ },
81
+ "gpu": "NVIDIA GeForce RTX 2070 SUPER",
82
+ "gpu_count": 1,
83
+ "gpu_devices": [
84
+ {
85
+ "name": "NVIDIA GeForce RTX 2070 SUPER",
86
+ "memory_total": 8589934592
87
+ }
88
+ ],
89
+ "memory": {
90
+ "total": 62.71596908569336
91
+ }
92
+ }
205123/wandb/run-20241210_205130-svysfc56/files/wandb-summary.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"eval/video": {"_type": "video-file", "sha256": "b6b0b3ba17b7513a5df9bca8084b0eda15d2f82df085783449312a47400bba10", "size": 942048, "path": "media/videos/eval/video_0_b6b0b3ba17b7513a5df9.mp4"}, "global_step": 109, "_timestamp": 1733881964.3609016, "_runtime": 74.04860258102417, "_step": 0, "_wandb": {"runtime": 73}}
205123/wandb/run-20241210_205130-svysfc56/logs/debug-internal.log ADDED
@@ -0,0 +1,351 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ 2024-12-10 20:51:30,311 INFO StreamThr :597014 [internal.py:wandb_internal():85] W&B internal server running at pid: 597014, started at: 2024-12-10 20:51:30.311157
2
+ 2024-12-10 20:51:30,312 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: status
3
+ 2024-12-10 20:51:30,325 INFO WriterThread:597014 [datastore.py:open_for_write():87] open: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/wandb/run-20241210_205130-svysfc56/run-svysfc56.wandb
4
+ 2024-12-10 20:51:30,325 DEBUG SenderThread:597014 [sender.py:send():379] send: header
5
+ 2024-12-10 20:51:30,336 DEBUG SenderThread:597014 [sender.py:send():379] send: run
6
+ 2024-12-10 20:51:30,645 INFO SenderThread:597014 [dir_watcher.py:__init__():211] watching files in: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/wandb/run-20241210_205130-svysfc56/files
7
+ 2024-12-10 20:51:30,645 INFO SenderThread:597014 [sender.py:_start_run_threads():1188] run started: svysfc56 with start time 1733881890.312299
8
+ 2024-12-10 20:51:30,656 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: check_version
9
+ 2024-12-10 20:51:30,656 DEBUG SenderThread:597014 [sender.py:send_request():406] send_request: check_version
10
+ 2024-12-10 20:51:30,743 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: run_start
11
+ 2024-12-10 20:51:30,748 DEBUG HandlerThread:597014 [system_info.py:__init__():26] System info init
12
+ 2024-12-10 20:51:30,748 DEBUG HandlerThread:597014 [system_info.py:__init__():41] System info init done
13
+ 2024-12-10 20:51:30,748 INFO HandlerThread:597014 [system_monitor.py:start():194] Starting system monitor
14
+ 2024-12-10 20:51:30,748 INFO SystemMonitor:597014 [system_monitor.py:_start():158] Starting system asset monitoring threads
15
+ 2024-12-10 20:51:30,749 INFO HandlerThread:597014 [system_monitor.py:probe():214] Collecting system info
16
+ 2024-12-10 20:51:30,749 INFO SystemMonitor:597014 [interfaces.py:start():188] Started cpu monitoring
17
+ 2024-12-10 20:51:30,750 INFO SystemMonitor:597014 [interfaces.py:start():188] Started disk monitoring
18
+ 2024-12-10 20:51:30,750 INFO SystemMonitor:597014 [interfaces.py:start():188] Started gpu monitoring
19
+ 2024-12-10 20:51:30,751 INFO SystemMonitor:597014 [interfaces.py:start():188] Started memory monitoring
20
+ 2024-12-10 20:51:30,751 INFO SystemMonitor:597014 [interfaces.py:start():188] Started network monitoring
21
+ 2024-12-10 20:51:30,761 DEBUG HandlerThread:597014 [system_info.py:probe():152] Probing system
22
+ 2024-12-10 20:51:30,762 DEBUG HandlerThread:597014 [system_info.py:_probe_git():137] Probing git
23
+ 2024-12-10 20:51:30,767 DEBUG HandlerThread:597014 [system_info.py:_probe_git():145] Probing git done
24
+ 2024-12-10 20:51:30,767 DEBUG HandlerThread:597014 [system_info.py:probe():200] Probing system done
25
+ 2024-12-10 20:51:30,767 DEBUG HandlerThread:597014 [system_monitor.py:probe():223] {'os': 'Linux-5.15.0-125-generic-x86_64-with-glibc2.31', 'python': '3.10.14', 'heartbeatAt': '2024-12-11T01:51:30.761221', 'startedAt': '2024-12-11T01:51:30.302896', 'docker': None, 'cuda': None, 'args': ('agent=diffusion', 'suite=frankagym', 'suite/frankagym_task@_global_=insertion'), 'state': 'running', 'program': '/home/leonmkim/fish_leon/FISH/eval_robot.py', 'codePathLocal': None, 'codePath': 'FISH/eval_robot.py', 'git': {'remote': 'https://github.com/leonmkim/fish_leon.git', 'commit': '4f7c0421eb84a7caf6bf5670f68777523362676f'}, 'email': 'leonmkim@seas.upenn.edu', 'root': '/home/leonmkim/fish_leon', 'host': 'leonmkim-ROG-Strix-G15CS-G15CS', 'username': 'leonmkim', 'executable': '/home/leonmkim/.pyenv/versions/lerobot/bin/python', 'cpu_count': 8, 'cpu_count_logical': 8, 'cpu_freq': {'current': 4124.7015, 'min': 800.0, 'max': 4700.0}, 'cpu_freq_per_core': [{'current': 3000.0, 'min': 800.0, 'max': 4700.0}, {'current': 4499.948, 'min': 800.0, 'max': 4700.0}, {'current': 4499.997, 'min': 800.0, 'max': 4700.0}, {'current': 3000.0, 'min': 800.0, 'max': 4700.0}, {'current': 3000.0, 'min': 800.0, 'max': 4700.0}, {'current': 3000.0, 'min': 800.0, 'max': 4700.0}, {'current': 3000.0, 'min': 800.0, 'max': 4700.0}, {'current': 4499.999, 'min': 800.0, 'max': 4700.0}], 'disk': {'/': {'total': 915.3232879638672, 'used': 364.1780319213867}}, 'gpu': 'NVIDIA GeForce RTX 2070 SUPER', 'gpu_count': 1, 'gpu_devices': [{'name': 'NVIDIA GeForce RTX 2070 SUPER', 'memory_total': 8589934592}], 'memory': {'total': 62.71596908569336}}
26
+ 2024-12-10 20:51:30,768 INFO HandlerThread:597014 [system_monitor.py:probe():224] Finished collecting system info
27
+ 2024-12-10 20:51:30,768 INFO HandlerThread:597014 [system_monitor.py:probe():227] Publishing system info
28
+ 2024-12-10 20:51:30,768 DEBUG HandlerThread:597014 [system_info.py:_save_code():44] Saving code
29
+ 2024-12-10 20:51:30,769 DEBUG HandlerThread:597014 [system_info.py:_save_code():67] Saving code done
30
+ 2024-12-10 20:51:30,769 DEBUG HandlerThread:597014 [system_info.py:_save_patches():84] Saving git patches
31
+ 2024-12-10 20:51:30,832 DEBUG HandlerThread:597014 [system_info.py:_save_patches():126] Saving git patches done
32
+ 2024-12-10 20:51:30,832 INFO HandlerThread:597014 [system_monitor.py:probe():229] Finished publishing system info
33
+ 2024-12-10 20:51:30,836 DEBUG SenderThread:597014 [sender.py:send():379] send: files
34
+ 2024-12-10 20:51:30,836 INFO SenderThread:597014 [sender.py:_save_file():1454] saving file wandb-metadata.json with policy now
35
+ 2024-12-10 20:51:30,836 INFO SenderThread:597014 [sender.py:_save_file():1454] saving file code/FISH/eval_robot.py with policy now
36
+ 2024-12-10 20:51:30,836 INFO SenderThread:597014 [sender.py:_save_file():1454] saving file diff.patch with policy now
37
+ 2024-12-10 20:51:31,124 INFO wandb-upload_2:597014 [upload_job.py:push():130] Uploaded file /tmp/tmp01kl0ofywandb/tiz5ae9w-diff.patch
38
+ 2024-12-10 20:51:31,126 INFO wandb-upload_0:597014 [upload_job.py:push():130] Uploaded file /tmp/tmp01kl0ofywandb/p6o167mp-wandb-metadata.json
39
+ 2024-12-10 20:51:31,143 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: python_packages
40
+ 2024-12-10 20:51:31,144 DEBUG SenderThread:597014 [sender.py:send_request():406] send_request: python_packages
41
+ 2024-12-10 20:51:31,145 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: stop_status
42
+ 2024-12-10 20:51:31,145 DEBUG SenderThread:597014 [sender.py:send_request():406] send_request: stop_status
43
+ 2024-12-10 20:51:31,146 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
44
+ 2024-12-10 20:51:31,147 INFO HandlerThread:597014 [handler.py:handle_tbrecord():783] handling tbrecord: tbrecord {
45
+ log_dir: "/home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/tb"
46
+ save: true
47
+ }
48
+ control {
49
+ relay_id: "42f7013eb6ef4a01b21dd778abc8c686"
50
+ }
51
+ _info {
52
+ stream_id: "svysfc56"
53
+ }
54
+
55
+ 2024-12-10 20:51:31,147 DEBUG HandlerThread:597014 [config_util.py:dict_from_config_file():70] no default config file found in config-defaults.yaml
56
+ 2024-12-10 20:51:31,173 INFO wandb-upload_1:597014 [upload_job.py:push():130] Uploaded file /tmp/tmp01kl0ofywandb/9ln43j5e-code/FISH/eval_robot.py
57
+ 2024-12-10 20:51:31,218 DEBUG SenderThread:597014 [sender.py:send():379] send: telemetry
58
+ 2024-12-10 20:51:31,218 DEBUG SenderThread:597014 [sender.py:send():379] send: tbrecord
59
+ 2024-12-10 20:51:31,218 DEBUG SenderThread:597014 [sender.py:send():379] send: files
60
+ 2024-12-10 20:51:31,218 INFO SenderThread:597014 [sender.py:_save_file():1454] saving file events.out.tfevents.1733881891.leonmkim-ROG-Strix-G15CS-G15CS.596892.0 with policy live
61
+ 2024-12-10 20:51:31,415 INFO wandb-upload_2:597014 [upload_job.py:push():130] Uploaded file /tmp/tmp01kl0ofywandb/08emxf8k-events.out.tfevents.1733881891.leonmkim-ROG-Strix-G15CS-G15CS.596892.0
62
+ 2024-12-10 20:51:31,647 INFO Thread-12 :597014 [dir_watcher.py:_on_file_created():271] file/dir created: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/wandb/run-20241210_205130-svysfc56/files/diff.patch
63
+ 2024-12-10 20:51:31,647 INFO Thread-12 :597014 [dir_watcher.py:_on_file_created():271] file/dir created: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/wandb/run-20241210_205130-svysfc56/files/code/FISH/eval_robot.py
64
+ 2024-12-10 20:51:31,647 INFO Thread-12 :597014 [dir_watcher.py:_on_file_created():271] file/dir created: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/wandb/run-20241210_205130-svysfc56/files/requirements.txt
65
+ 2024-12-10 20:51:31,647 INFO Thread-12 :597014 [dir_watcher.py:_on_file_created():271] file/dir created: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/wandb/run-20241210_205130-svysfc56/files/output.log
66
+ 2024-12-10 20:51:31,647 INFO Thread-12 :597014 [dir_watcher.py:_on_file_created():271] file/dir created: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/wandb/run-20241210_205130-svysfc56/files/wandb-metadata.json
67
+ 2024-12-10 20:51:31,647 INFO Thread-12 :597014 [dir_watcher.py:_on_file_created():271] file/dir created: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/wandb/run-20241210_205130-svysfc56/files/events.out.tfevents.1733881891.leonmkim-ROG-Strix-G15CS-G15CS.596892.0
68
+ 2024-12-10 20:51:31,647 INFO Thread-12 :597014 [dir_watcher.py:_on_file_created():271] file/dir created: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/wandb/run-20241210_205130-svysfc56/files/code/FISH
69
+ 2024-12-10 20:51:31,647 INFO Thread-12 :597014 [dir_watcher.py:_on_file_created():271] file/dir created: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/wandb/run-20241210_205130-svysfc56/files/code
70
+ 2024-12-10 20:51:32,144 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
71
+ 2024-12-10 20:51:33,145 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
72
+ 2024-12-10 20:51:33,646 INFO Thread-12 :597014 [dir_watcher.py:_on_file_modified():288] file/dir modified: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/wandb/run-20241210_205130-svysfc56/files/output.log
73
+ 2024-12-10 20:51:34,145 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
74
+ 2024-12-10 20:51:35,145 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
75
+ 2024-12-10 20:51:36,144 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: status_report
76
+ 2024-12-10 20:51:36,145 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
77
+ 2024-12-10 20:51:37,145 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
78
+ 2024-12-10 20:51:38,145 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
79
+ 2024-12-10 20:51:39,145 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
80
+ 2024-12-10 20:51:40,145 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
81
+ 2024-12-10 20:51:41,144 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: status_report
82
+ 2024-12-10 20:51:41,146 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
83
+ 2024-12-10 20:51:42,146 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
84
+ 2024-12-10 20:51:43,146 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
85
+ 2024-12-10 20:51:44,146 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
86
+ 2024-12-10 20:51:45,146 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
87
+ 2024-12-10 20:51:46,144 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: stop_status
88
+ 2024-12-10 20:51:46,144 DEBUG SenderThread:597014 [sender.py:send_request():406] send_request: stop_status
89
+ 2024-12-10 20:51:46,186 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
90
+ 2024-12-10 20:51:46,300 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: status_report
91
+ 2024-12-10 20:51:46,908 DEBUG SenderThread:597014 [sender.py:send():379] send: config
92
+ 2024-12-10 20:51:47,146 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
93
+ 2024-12-10 20:51:48,146 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
94
+ 2024-12-10 20:51:49,146 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
95
+ 2024-12-10 20:51:49,651 INFO Thread-12 :597014 [dir_watcher.py:_on_file_modified():288] file/dir modified: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/wandb/run-20241210_205130-svysfc56/files/output.log
96
+ 2024-12-10 20:51:50,146 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
97
+ 2024-12-10 20:51:51,147 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
98
+ 2024-12-10 20:51:51,335 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: status_report
99
+ 2024-12-10 20:51:52,147 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
100
+ 2024-12-10 20:51:53,147 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
101
+ 2024-12-10 20:51:53,652 INFO Thread-12 :597014 [dir_watcher.py:_on_file_modified():288] file/dir modified: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/wandb/run-20241210_205130-svysfc56/files/output.log
102
+ 2024-12-10 20:51:54,147 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
103
+ 2024-12-10 20:51:55,147 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
104
+ 2024-12-10 20:51:56,147 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
105
+ 2024-12-10 20:51:57,145 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: status_report
106
+ 2024-12-10 20:51:57,147 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
107
+ 2024-12-10 20:51:57,654 INFO Thread-12 :597014 [dir_watcher.py:_on_file_modified():288] file/dir modified: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/wandb/run-20241210_205130-svysfc56/files/output.log
108
+ 2024-12-10 20:51:58,148 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
109
+ 2024-12-10 20:51:59,148 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
110
+ 2024-12-10 20:52:00,148 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
111
+ 2024-12-10 20:52:01,144 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: stop_status
112
+ 2024-12-10 20:52:01,144 DEBUG SenderThread:597014 [sender.py:send_request():406] send_request: stop_status
113
+ 2024-12-10 20:52:01,189 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
114
+ 2024-12-10 20:52:02,148 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
115
+ 2024-12-10 20:52:02,300 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: status_report
116
+ 2024-12-10 20:52:02,655 INFO Thread-12 :597014 [dir_watcher.py:_on_file_modified():288] file/dir modified: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/wandb/run-20241210_205130-svysfc56/files/config.yaml
117
+ 2024-12-10 20:52:03,148 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
118
+ 2024-12-10 20:52:04,149 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
119
+ 2024-12-10 20:52:05,149 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
120
+ 2024-12-10 20:52:06,182 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
121
+ 2024-12-10 20:52:07,149 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
122
+ 2024-12-10 20:52:07,657 INFO Thread-12 :597014 [dir_watcher.py:_on_file_modified():288] file/dir modified: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/wandb/run-20241210_205130-svysfc56/files/output.log
123
+ 2024-12-10 20:52:08,149 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
124
+ 2024-12-10 20:52:08,194 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: status_report
125
+ 2024-12-10 20:52:09,149 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
126
+ 2024-12-10 20:52:09,657 INFO Thread-12 :597014 [dir_watcher.py:_on_file_modified():288] file/dir modified: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/wandb/run-20241210_205130-svysfc56/files/output.log
127
+ 2024-12-10 20:52:10,149 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
128
+ 2024-12-10 20:52:11,149 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
129
+ 2024-12-10 20:52:11,658 INFO Thread-12 :597014 [dir_watcher.py:_on_file_modified():288] file/dir modified: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/wandb/run-20241210_205130-svysfc56/files/output.log
130
+ 2024-12-10 20:52:12,149 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
131
+ 2024-12-10 20:52:13,149 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
132
+ 2024-12-10 20:52:13,219 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: status_report
133
+ 2024-12-10 20:52:13,659 INFO Thread-12 :597014 [dir_watcher.py:_on_file_modified():288] file/dir modified: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/wandb/run-20241210_205130-svysfc56/files/output.log
134
+ 2024-12-10 20:52:14,150 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
135
+ 2024-12-10 20:52:15,150 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
136
+ 2024-12-10 20:52:15,659 INFO Thread-12 :597014 [dir_watcher.py:_on_file_modified():288] file/dir modified: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/wandb/run-20241210_205130-svysfc56/files/output.log
137
+ 2024-12-10 20:52:16,144 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: stop_status
138
+ 2024-12-10 20:52:16,144 DEBUG SenderThread:597014 [sender.py:send_request():406] send_request: stop_status
139
+ 2024-12-10 20:52:16,150 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
140
+ 2024-12-10 20:52:17,150 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
141
+ 2024-12-10 20:52:17,660 INFO Thread-12 :597014 [dir_watcher.py:_on_file_modified():288] file/dir modified: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/wandb/run-20241210_205130-svysfc56/files/output.log
142
+ 2024-12-10 20:52:18,150 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
143
+ 2024-12-10 20:52:18,404 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: status_report
144
+ 2024-12-10 20:52:19,152 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
145
+ 2024-12-10 20:52:19,661 INFO Thread-12 :597014 [dir_watcher.py:_on_file_modified():288] file/dir modified: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/wandb/run-20241210_205130-svysfc56/files/output.log
146
+ 2024-12-10 20:52:20,152 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
147
+ 2024-12-10 20:52:21,152 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
148
+ 2024-12-10 20:52:21,661 INFO Thread-12 :597014 [dir_watcher.py:_on_file_modified():288] file/dir modified: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/wandb/run-20241210_205130-svysfc56/files/output.log
149
+ 2024-12-10 20:52:22,152 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
150
+ 2024-12-10 20:52:23,152 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
151
+ 2024-12-10 20:52:23,407 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: status_report
152
+ 2024-12-10 20:52:23,662 INFO Thread-12 :597014 [dir_watcher.py:_on_file_modified():288] file/dir modified: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/wandb/run-20241210_205130-svysfc56/files/output.log
153
+ 2024-12-10 20:52:24,152 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
154
+ 2024-12-10 20:52:25,153 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
155
+ 2024-12-10 20:52:25,662 INFO Thread-12 :597014 [dir_watcher.py:_on_file_modified():288] file/dir modified: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/wandb/run-20241210_205130-svysfc56/files/output.log
156
+ 2024-12-10 20:52:26,153 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
157
+ 2024-12-10 20:52:27,153 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
158
+ 2024-12-10 20:52:27,663 INFO Thread-12 :597014 [dir_watcher.py:_on_file_modified():288] file/dir modified: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/wandb/run-20241210_205130-svysfc56/files/output.log
159
+ 2024-12-10 20:52:28,153 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
160
+ 2024-12-10 20:52:28,607 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: status_report
161
+ 2024-12-10 20:52:29,153 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
162
+ 2024-12-10 20:52:29,664 INFO Thread-12 :597014 [dir_watcher.py:_on_file_modified():288] file/dir modified: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/wandb/run-20241210_205130-svysfc56/files/output.log
163
+ 2024-12-10 20:52:30,154 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
164
+ 2024-12-10 20:52:30,752 DEBUG SystemMonitor:597014 [system_monitor.py:_start():172] Starting system metrics aggregation loop
165
+ 2024-12-10 20:52:30,752 DEBUG SenderThread:597014 [sender.py:send():379] send: stats
166
+ 2024-12-10 20:52:31,144 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: stop_status
167
+ 2024-12-10 20:52:31,144 DEBUG SenderThread:597014 [sender.py:send_request():406] send_request: stop_status
168
+ 2024-12-10 20:52:31,154 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
169
+ 2024-12-10 20:52:31,666 INFO Thread-12 :597014 [dir_watcher.py:_on_file_modified():288] file/dir modified: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/wandb/run-20241210_205130-svysfc56/files/output.log
170
+ 2024-12-10 20:52:32,154 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
171
+ 2024-12-10 20:52:33,154 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
172
+ 2024-12-10 20:52:33,667 INFO Thread-12 :597014 [dir_watcher.py:_on_file_modified():288] file/dir modified: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/wandb/run-20241210_205130-svysfc56/files/output.log
173
+ 2024-12-10 20:52:33,908 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: status_report
174
+ 2024-12-10 20:52:34,154 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
175
+ 2024-12-10 20:52:34,667 INFO Thread-12 :597014 [dir_watcher.py:_on_file_modified():288] file/dir modified: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/wandb/run-20241210_205130-svysfc56/files/output.log
176
+ 2024-12-10 20:52:35,154 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
177
+ 2024-12-10 20:52:35,668 INFO Thread-12 :597014 [dir_watcher.py:_on_file_modified():288] file/dir modified: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/wandb/run-20241210_205130-svysfc56/files/output.log
178
+ 2024-12-10 20:52:36,154 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
179
+ 2024-12-10 20:52:37,155 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
180
+ 2024-12-10 20:52:38,155 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
181
+ 2024-12-10 20:52:38,668 INFO Thread-12 :597014 [dir_watcher.py:_on_file_modified():288] file/dir modified: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/wandb/run-20241210_205130-svysfc56/files/output.log
182
+ 2024-12-10 20:52:39,155 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
183
+ 2024-12-10 20:52:39,371 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: status_report
184
+ 2024-12-10 20:52:40,155 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
185
+ 2024-12-10 20:52:41,155 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
186
+ 2024-12-10 20:52:42,155 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
187
+ 2024-12-10 20:52:43,155 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
188
+ 2024-12-10 20:52:44,155 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
189
+ 2024-12-10 20:52:44,374 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: partial_history
190
+ 2024-12-10 20:52:44,375 DEBUG SenderThread:597014 [sender.py:send():379] send: files
191
+ 2024-12-10 20:52:44,375 INFO SenderThread:597014 [sender.py:_save_file():1454] saving file media/videos/eval/video_0_b6b0b3ba17b7513a5df9.mp4 with policy now
192
+ 2024-12-10 20:52:44,375 DEBUG SenderThread:597014 [sender.py:send():379] send: history
193
+ 2024-12-10 20:52:44,375 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: status_report
194
+ 2024-12-10 20:52:44,375 DEBUG SenderThread:597014 [sender.py:send_request():406] send_request: summary_record
195
+ 2024-12-10 20:52:44,377 INFO SenderThread:597014 [sender.py:_save_file():1454] saving file wandb-summary.json with policy end
196
+ 2024-12-10 20:52:44,377 DEBUG SenderThread:597014 [sender.py:send():379] send: exit
197
+ 2024-12-10 20:52:44,377 INFO SenderThread:597014 [sender.py:send_exit():586] handling exit code: 0
198
+ 2024-12-10 20:52:44,377 INFO SenderThread:597014 [sender.py:send_exit():588] handling runtime: 73
199
+ 2024-12-10 20:52:44,377 INFO SenderThread:597014 [sender.py:_save_file():1454] saving file wandb-summary.json with policy end
200
+ 2024-12-10 20:52:44,377 INFO SenderThread:597014 [sender.py:send_exit():594] send defer
201
+ 2024-12-10 20:52:44,379 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: defer
202
+ 2024-12-10 20:52:44,379 INFO HandlerThread:597014 [handler.py:handle_request_defer():184] handle defer: 0
203
+ 2024-12-10 20:52:44,380 DEBUG SenderThread:597014 [sender.py:send_request():406] send_request: defer
204
+ 2024-12-10 20:52:44,380 INFO SenderThread:597014 [sender.py:send_request_defer():610] handle sender defer: 0
205
+ 2024-12-10 20:52:44,380 INFO SenderThread:597014 [sender.py:transition_state():614] send defer: 1
206
+ 2024-12-10 20:52:44,380 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: defer
207
+ 2024-12-10 20:52:44,380 INFO HandlerThread:597014 [handler.py:handle_request_defer():184] handle defer: 1
208
+ 2024-12-10 20:52:44,380 DEBUG SenderThread:597014 [sender.py:send_request():406] send_request: defer
209
+ 2024-12-10 20:52:44,380 INFO SenderThread:597014 [sender.py:send_request_defer():610] handle sender defer: 1
210
+ 2024-12-10 20:52:44,380 INFO SenderThread:597014 [sender.py:transition_state():614] send defer: 2
211
+ 2024-12-10 20:52:44,380 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: defer
212
+ 2024-12-10 20:52:44,380 INFO HandlerThread:597014 [handler.py:handle_request_defer():184] handle defer: 2
213
+ 2024-12-10 20:52:44,380 INFO HandlerThread:597014 [system_monitor.py:finish():203] Stopping system monitor
214
+ 2024-12-10 20:52:44,381 DEBUG SystemMonitor:597014 [system_monitor.py:_start():179] Finished system metrics aggregation loop
215
+ 2024-12-10 20:52:44,381 DEBUG SystemMonitor:597014 [system_monitor.py:_start():183] Publishing last batch of metrics
216
+ 2024-12-10 20:52:44,381 INFO HandlerThread:597014 [interfaces.py:finish():200] Joined cpu monitor
217
+ 2024-12-10 20:52:44,382 INFO HandlerThread:597014 [interfaces.py:finish():200] Joined disk monitor
218
+ 2024-12-10 20:52:44,439 INFO HandlerThread:597014 [interfaces.py:finish():200] Joined gpu monitor
219
+ 2024-12-10 20:52:44,439 INFO HandlerThread:597014 [interfaces.py:finish():200] Joined memory monitor
220
+ 2024-12-10 20:52:44,439 INFO HandlerThread:597014 [interfaces.py:finish():200] Joined network monitor
221
+ 2024-12-10 20:52:44,440 DEBUG SenderThread:597014 [sender.py:send_request():406] send_request: defer
222
+ 2024-12-10 20:52:44,440 INFO SenderThread:597014 [sender.py:send_request_defer():610] handle sender defer: 2
223
+ 2024-12-10 20:52:44,440 INFO SenderThread:597014 [sender.py:transition_state():614] send defer: 3
224
+ 2024-12-10 20:52:44,440 DEBUG SenderThread:597014 [sender.py:send():379] send: stats
225
+ 2024-12-10 20:52:44,440 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: defer
226
+ 2024-12-10 20:52:44,440 INFO HandlerThread:597014 [handler.py:handle_request_defer():184] handle defer: 3
227
+ 2024-12-10 20:52:44,440 DEBUG SenderThread:597014 [sender.py:send_request():406] send_request: defer
228
+ 2024-12-10 20:52:44,440 INFO SenderThread:597014 [sender.py:send_request_defer():610] handle sender defer: 3
229
+ 2024-12-10 20:52:44,440 INFO SenderThread:597014 [sender.py:transition_state():614] send defer: 4
230
+ 2024-12-10 20:52:44,441 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: defer
231
+ 2024-12-10 20:52:44,441 INFO HandlerThread:597014 [handler.py:handle_request_defer():184] handle defer: 4
232
+ 2024-12-10 20:52:44,670 INFO Thread-12 :597014 [dir_watcher.py:_on_file_created():271] file/dir created: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/wandb/run-20241210_205130-svysfc56/files/wandb-summary.json
233
+ 2024-12-10 20:52:44,671 INFO Thread-12 :597014 [dir_watcher.py:_on_file_created():271] file/dir created: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/wandb/run-20241210_205130-svysfc56/files/media/videos/eval/video_0_b6b0b3ba17b7513a5df9.mp4
234
+ 2024-12-10 20:52:44,671 INFO Thread-12 :597014 [dir_watcher.py:_on_file_created():271] file/dir created: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/wandb/run-20241210_205130-svysfc56/files/media/videos/eval
235
+ 2024-12-10 20:52:44,671 INFO Thread-12 :597014 [dir_watcher.py:_on_file_created():271] file/dir created: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/wandb/run-20241210_205130-svysfc56/files/media/videos
236
+ 2024-12-10 20:52:44,671 INFO Thread-12 :597014 [dir_watcher.py:_on_file_created():271] file/dir created: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/wandb/run-20241210_205130-svysfc56/files/media
237
+ 2024-12-10 20:52:44,696 INFO wandb-upload_1:597014 [upload_job.py:push():130] Uploaded file /tmp/tmp01kl0ofywandb/6w704qfx-media/videos/eval/video_0_b6b0b3ba17b7513a5df9.mp4
238
+ 2024-12-10 20:52:45,030 INFO wandb-upload_2:597014 [upload_job.py:push():130] Uploaded file /tmp/tmp01kl0ofywandb/kk0w0e5k-media/videos/eval/video_0_b6b0b3ba17b7513a5df9.mp4
239
+ 2024-12-10 20:52:45,671 INFO Thread-12 :597014 [dir_watcher.py:_on_file_modified():288] file/dir modified: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/wandb/run-20241210_205130-svysfc56/files/output.log
240
+ 2024-12-10 20:52:51,156 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: poll_exit
241
+ 2024-12-10 20:52:51,156 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: status_report
242
+ 2024-12-10 20:52:51,156 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: keepalive
243
+ 2024-12-10 20:52:51,157 DEBUG SenderThread:597014 [sender.py:send_request():406] send_request: defer
244
+ 2024-12-10 20:52:51,157 INFO SenderThread:597014 [sender.py:send_request_defer():610] handle sender defer: 4
245
+ 2024-12-10 20:52:51,157 INFO SenderThread:597014 [sender.py:transition_state():614] send defer: 5
246
+ 2024-12-10 20:52:51,157 DEBUG SenderThread:597014 [sender.py:send_request():406] send_request: poll_exit
247
+ 2024-12-10 20:52:51,157 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: defer
248
+ 2024-12-10 20:52:51,157 INFO HandlerThread:597014 [handler.py:handle_request_defer():184] handle defer: 5
249
+ 2024-12-10 20:52:51,157 DEBUG SenderThread:597014 [sender.py:send():379] send: summary
250
+ 2024-12-10 20:52:51,158 INFO SenderThread:597014 [sender.py:_save_file():1454] saving file wandb-summary.json with policy end
251
+ 2024-12-10 20:52:51,158 DEBUG SenderThread:597014 [sender.py:send_request():406] send_request: defer
252
+ 2024-12-10 20:52:51,158 INFO SenderThread:597014 [sender.py:send_request_defer():610] handle sender defer: 5
253
+ 2024-12-10 20:52:51,158 INFO SenderThread:597014 [sender.py:transition_state():614] send defer: 6
254
+ 2024-12-10 20:52:51,158 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: defer
255
+ 2024-12-10 20:52:51,158 INFO HandlerThread:597014 [handler.py:handle_request_defer():184] handle defer: 6
256
+ 2024-12-10 20:52:51,158 DEBUG SenderThread:597014 [sender.py:send_request():406] send_request: defer
257
+ 2024-12-10 20:52:51,158 INFO SenderThread:597014 [sender.py:send_request_defer():610] handle sender defer: 6
258
+ 2024-12-10 20:52:51,158 INFO SenderThread:597014 [sender.py:transition_state():614] send defer: 7
259
+ 2024-12-10 20:52:51,158 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: status_report
260
+ 2024-12-10 20:52:51,158 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: defer
261
+ 2024-12-10 20:52:51,158 INFO HandlerThread:597014 [handler.py:handle_request_defer():184] handle defer: 7
262
+ 2024-12-10 20:52:51,159 DEBUG SenderThread:597014 [sender.py:send_request():406] send_request: defer
263
+ 2024-12-10 20:52:51,159 INFO SenderThread:597014 [sender.py:send_request_defer():610] handle sender defer: 7
264
+ 2024-12-10 20:52:51,375 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: poll_exit
265
+ 2024-12-10 20:52:51,673 INFO Thread-12 :597014 [dir_watcher.py:_on_file_modified():288] file/dir modified: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/wandb/run-20241210_205130-svysfc56/files/wandb-summary.json
266
+ 2024-12-10 20:52:51,921 INFO SenderThread:597014 [sender.py:transition_state():614] send defer: 8
267
+ 2024-12-10 20:52:51,922 DEBUG SenderThread:597014 [sender.py:send_request():406] send_request: poll_exit
268
+ 2024-12-10 20:52:51,922 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: defer
269
+ 2024-12-10 20:52:51,922 INFO HandlerThread:597014 [handler.py:handle_request_defer():184] handle defer: 8
270
+ 2024-12-10 20:52:51,922 DEBUG SenderThread:597014 [sender.py:send_request():406] send_request: defer
271
+ 2024-12-10 20:52:51,922 INFO SenderThread:597014 [sender.py:send_request_defer():610] handle sender defer: 8
272
+ 2024-12-10 20:52:51,922 INFO SenderThread:597014 [job_builder.py:build():440] Attempting to build job artifact
273
+ 2024-12-10 20:52:51,924 INFO SenderThread:597014 [job_builder.py:_get_source_type():569] is repo sourced job
274
+ 2024-12-10 20:52:51,942 INFO SenderThread:597014 [job_builder.py:build():545] adding wandb-job metadata file
275
+ 2024-12-10 20:52:51,951 INFO SenderThread:597014 [sender.py:transition_state():614] send defer: 9
276
+ 2024-12-10 20:52:51,952 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: defer
277
+ 2024-12-10 20:52:51,952 INFO HandlerThread:597014 [handler.py:handle_request_defer():184] handle defer: 9
278
+ 2024-12-10 20:52:51,952 DEBUG SenderThread:597014 [sender.py:send():379] send: artifact
279
+ 2024-12-10 20:52:52,376 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: poll_exit
280
+ 2024-12-10 20:52:52,626 INFO wandb-upload_0:597014 [upload_job.py:push():85] Skipped uploading /home/leonmkim/.local/share/wandb/artifacts/staging/tmptoffctew
281
+ 2024-12-10 20:52:52,673 INFO Thread-12 :597014 [dir_watcher.py:_on_file_modified():288] file/dir modified: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/wandb/run-20241210_205130-svysfc56/files/output.log
282
+ 2024-12-10 20:52:52,858 INFO wandb-upload_2:597014 [upload_job.py:push():88] Uploaded file /tmp/tmpemx2d4yl/wandb-job.json
283
+ 2024-12-10 20:52:52,880 INFO wandb-upload_1:597014 [upload_job.py:push():88] Uploaded file /home/leonmkim/.local/share/wandb/artifacts/staging/tmpc2y7tffk
284
+ 2024-12-10 20:52:53,418 INFO SenderThread:597014 [sender.py:send_artifact():1537] sent artifact job-https___github.com_leonmkim_fish_leon.git_FISH_eval_robot.py - {'id': 'QXJ0aWZhY3Q6MTM2Njc1OTg1Ng==', 'state': 'PENDING', 'artifactSequence': {'id': 'QXJ0aWZhY3RDb2xsZWN0aW9uOjMxNzk5NjkyNQ==', 'latestArtifact': {'id': 'QXJ0aWZhY3Q6MTM1MTcyNjQ1OA==', 'versionIndex': 335}}}
285
+ 2024-12-10 20:52:53,418 DEBUG SenderThread:597014 [sender.py:send_request():406] send_request: defer
286
+ 2024-12-10 20:52:53,418 INFO SenderThread:597014 [sender.py:send_request_defer():610] handle sender defer: 9
287
+ 2024-12-10 20:52:53,418 INFO SenderThread:597014 [dir_watcher.py:finish():358] shutting down directory watcher
288
+ 2024-12-10 20:52:53,674 INFO SenderThread:597014 [dir_watcher.py:finish():388] scan: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/wandb/run-20241210_205130-svysfc56/files
289
+ 2024-12-10 20:52:53,674 INFO SenderThread:597014 [dir_watcher.py:finish():402] scan save: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/wandb/run-20241210_205130-svysfc56/files/config.yaml config.yaml
290
+ 2024-12-10 20:52:53,674 INFO SenderThread:597014 [dir_watcher.py:finish():402] scan save: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/wandb/run-20241210_205130-svysfc56/files/wandb-summary.json wandb-summary.json
291
+ 2024-12-10 20:52:53,674 INFO SenderThread:597014 [dir_watcher.py:finish():402] scan save: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/wandb/run-20241210_205130-svysfc56/files/requirements.txt requirements.txt
292
+ 2024-12-10 20:52:53,674 INFO SenderThread:597014 [dir_watcher.py:finish():402] scan save: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/wandb/run-20241210_205130-svysfc56/files/wandb-metadata.json wandb-metadata.json
293
+ 2024-12-10 20:52:53,674 INFO SenderThread:597014 [dir_watcher.py:finish():402] scan save: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/wandb/run-20241210_205130-svysfc56/files/events.out.tfevents.1733881891.leonmkim-ROG-Strix-G15CS-G15CS.596892.0 events.out.tfevents.1733881891.leonmkim-ROG-Strix-G15CS-G15CS.596892.0
294
+ 2024-12-10 20:52:53,674 INFO SenderThread:597014 [dir_watcher.py:finish():402] scan save: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/wandb/run-20241210_205130-svysfc56/files/output.log output.log
295
+ 2024-12-10 20:52:53,676 INFO SenderThread:597014 [dir_watcher.py:finish():402] scan save: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/wandb/run-20241210_205130-svysfc56/files/diff.patch diff.patch
296
+ 2024-12-10 20:52:53,680 INFO SenderThread:597014 [dir_watcher.py:finish():402] scan save: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/wandb/run-20241210_205130-svysfc56/files/code/FISH/eval_robot.py code/FISH/eval_robot.py
297
+ 2024-12-10 20:52:53,680 INFO SenderThread:597014 [dir_watcher.py:finish():402] scan save: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/wandb/run-20241210_205130-svysfc56/files/media/videos/eval/video_0_b6b0b3ba17b7513a5df9.mp4 media/videos/eval/video_0_b6b0b3ba17b7513a5df9.mp4
298
+ 2024-12-10 20:52:53,680 INFO SenderThread:597014 [sender.py:transition_state():614] send defer: 10
299
+ 2024-12-10 20:52:53,680 DEBUG SenderThread:597014 [sender.py:send_request():406] send_request: poll_exit
300
+ 2024-12-10 20:52:53,681 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: defer
301
+ 2024-12-10 20:52:53,681 INFO HandlerThread:597014 [handler.py:handle_request_defer():184] handle defer: 10
302
+ 2024-12-10 20:52:53,681 DEBUG SenderThread:597014 [sender.py:send_request():406] send_request: defer
303
+ 2024-12-10 20:52:53,681 INFO SenderThread:597014 [sender.py:send_request_defer():610] handle sender defer: 10
304
+ 2024-12-10 20:52:53,681 INFO SenderThread:597014 [file_pusher.py:finish():169] shutting down file pusher
305
+ 2024-12-10 20:52:53,872 INFO wandb-upload_2:597014 [upload_job.py:push():130] Uploaded file /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/wandb/run-20241210_205130-svysfc56/files/requirements.txt
306
+ 2024-12-10 20:52:53,888 INFO wandb-upload_0:597014 [upload_job.py:push():130] Uploaded file /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/wandb/run-20241210_205130-svysfc56/files/config.yaml
307
+ 2024-12-10 20:52:53,899 INFO wandb-upload_3:597014 [upload_job.py:push():130] Uploaded file /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/wandb/run-20241210_205130-svysfc56/files/wandb-summary.json
308
+ 2024-12-10 20:52:53,910 INFO wandb-upload_1:597014 [upload_job.py:push():130] Uploaded file /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/wandb/run-20241210_205130-svysfc56/files/output.log
309
+ 2024-12-10 20:52:54,110 INFO Thread-11 (_thread_body):597014 [sender.py:transition_state():614] send defer: 11
310
+ 2024-12-10 20:52:54,110 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: defer
311
+ 2024-12-10 20:52:54,111 INFO HandlerThread:597014 [handler.py:handle_request_defer():184] handle defer: 11
312
+ 2024-12-10 20:52:54,111 DEBUG SenderThread:597014 [sender.py:send_request():406] send_request: defer
313
+ 2024-12-10 20:52:54,111 INFO SenderThread:597014 [sender.py:send_request_defer():610] handle sender defer: 11
314
+ 2024-12-10 20:52:54,111 INFO SenderThread:597014 [file_pusher.py:join():175] waiting for file pusher
315
+ 2024-12-10 20:52:54,111 INFO SenderThread:597014 [sender.py:transition_state():614] send defer: 12
316
+ 2024-12-10 20:52:54,111 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: defer
317
+ 2024-12-10 20:52:54,111 INFO HandlerThread:597014 [handler.py:handle_request_defer():184] handle defer: 12
318
+ 2024-12-10 20:52:54,112 DEBUG SenderThread:597014 [sender.py:send_request():406] send_request: defer
319
+ 2024-12-10 20:52:54,112 INFO SenderThread:597014 [sender.py:send_request_defer():610] handle sender defer: 12
320
+ 2024-12-10 20:52:54,112 INFO SenderThread:597014 [file_stream.py:finish():601] file stream finish called
321
+ 2024-12-10 20:52:54,317 INFO SenderThread:597014 [file_stream.py:finish():605] file stream finish is done
322
+ 2024-12-10 20:52:54,317 INFO SenderThread:597014 [sender.py:transition_state():614] send defer: 13
323
+ 2024-12-10 20:52:54,318 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: defer
324
+ 2024-12-10 20:52:54,318 INFO HandlerThread:597014 [handler.py:handle_request_defer():184] handle defer: 13
325
+ 2024-12-10 20:52:54,318 DEBUG SenderThread:597014 [sender.py:send_request():406] send_request: defer
326
+ 2024-12-10 20:52:54,318 INFO SenderThread:597014 [sender.py:send_request_defer():610] handle sender defer: 13
327
+ 2024-12-10 20:52:54,318 INFO SenderThread:597014 [sender.py:transition_state():614] send defer: 14
328
+ 2024-12-10 20:52:54,318 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: defer
329
+ 2024-12-10 20:52:54,318 INFO HandlerThread:597014 [handler.py:handle_request_defer():184] handle defer: 14
330
+ 2024-12-10 20:52:54,319 DEBUG SenderThread:597014 [sender.py:send():379] send: final
331
+ 2024-12-10 20:52:54,319 DEBUG SenderThread:597014 [sender.py:send():379] send: footer
332
+ 2024-12-10 20:52:54,319 DEBUG SenderThread:597014 [sender.py:send_request():406] send_request: defer
333
+ 2024-12-10 20:52:54,319 INFO SenderThread:597014 [sender.py:send_request_defer():610] handle sender defer: 14
334
+ 2024-12-10 20:52:54,319 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: poll_exit
335
+ 2024-12-10 20:52:54,320 DEBUG SenderThread:597014 [sender.py:send_request():406] send_request: poll_exit
336
+ 2024-12-10 20:52:54,320 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: server_info
337
+ 2024-12-10 20:52:54,320 DEBUG SenderThread:597014 [sender.py:send_request():406] send_request: server_info
338
+ 2024-12-10 20:52:54,322 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: poll_exit
339
+ 2024-12-10 20:52:54,322 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: get_summary
340
+ 2024-12-10 20:52:54,322 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: sampled_history
341
+ 2024-12-10 20:52:54,322 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: internal_messages
342
+ 2024-12-10 20:52:54,378 DEBUG SenderThread:597014 [sender.py:send_request():406] send_request: poll_exit
343
+ 2024-12-10 20:52:54,378 INFO MainThread:597014 [wandb_run.py:_footer_history_summary_info():4016] rendering history
344
+ 2024-12-10 20:52:54,378 INFO MainThread:597014 [wandb_run.py:_footer_history_summary_info():4048] rendering summary
345
+ 2024-12-10 20:52:54,379 INFO MainThread:597014 [wandb_run.py:_footer_sync_info():3975] logging synced files
346
+ 2024-12-10 20:52:54,379 DEBUG HandlerThread:597014 [handler.py:handle_request():158] handle_request: shutdown
347
+ 2024-12-10 20:52:54,379 INFO HandlerThread:597014 [handler.py:finish():882] shutting down handler
348
+ 2024-12-10 20:52:55,322 INFO WriterThread:597014 [datastore.py:close():296] close: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/wandb/run-20241210_205130-svysfc56/run-svysfc56.wandb
349
+ 2024-12-10 20:52:55,378 INFO SenderThread:597014 [sender.py:finish():1615] shutting down sender
350
+ 2024-12-10 20:52:55,378 INFO SenderThread:597014 [file_pusher.py:finish():169] shutting down file pusher
351
+ 2024-12-10 20:52:55,378 INFO SenderThread:597014 [file_pusher.py:join():175] waiting for file pusher
205123/wandb/run-20241210_205130-svysfc56/logs/debug.log ADDED
@@ -0,0 +1,31 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ 2024-12-10 20:51:30,303 INFO MainThread:596892 [wandb_setup.py:_flush():76] Current SDK version is 0.17.5
2
+ 2024-12-10 20:51:30,303 INFO MainThread:596892 [wandb_setup.py:_flush():76] Configure stats pid to 596892
3
+ 2024-12-10 20:51:30,303 INFO MainThread:596892 [wandb_setup.py:_flush():76] Loading settings from /home/leonmkim/.config/wandb/settings
4
+ 2024-12-10 20:51:30,303 INFO MainThread:596892 [wandb_setup.py:_flush():76] Loading settings from /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/wandb/settings
5
+ 2024-12-10 20:51:30,303 INFO MainThread:596892 [wandb_setup.py:_flush():76] Loading settings from environment variables: {}
6
+ 2024-12-10 20:51:30,303 INFO MainThread:596892 [wandb_setup.py:_flush():76] Applying setup settings: {'_disable_service': False}
7
+ 2024-12-10 20:51:30,304 INFO MainThread:596892 [wandb_setup.py:_flush():76] Inferring run settings from compute environment: {'program_relpath': 'FISH/eval_robot.py', 'program_abspath': '/home/leonmkim/fish_leon/FISH/eval_robot.py', 'program': '/home/leonmkim/fish_leon/FISH/eval_robot.py'}
8
+ 2024-12-10 20:51:30,304 INFO MainThread:596892 [wandb_setup.py:_flush():76] Applying login settings: {}
9
+ 2024-12-10 20:51:30,304 INFO MainThread:596892 [wandb_init.py:_log_setup():529] Logging user logs to /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/wandb/run-20241210_205130-svysfc56/logs/debug.log
10
+ 2024-12-10 20:51:30,304 INFO MainThread:596892 [wandb_init.py:_log_setup():530] Logging internal logs to /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/wandb/run-20241210_205130-svysfc56/logs/debug-internal.log
11
+ 2024-12-10 20:51:30,304 INFO MainThread:596892 [wandb_init.py:init():569] calling init triggers
12
+ 2024-12-10 20:51:30,304 INFO MainThread:596892 [wandb_init.py:init():576] wandb.init called with sweep_config: {}
13
+ config: {'root_dir': '/home/leonmkim/fish_leon', 'replay_buffer_size': 150000, 'replay_buffer_num_workers': 2, 'nstep': 3, 'batch_size': 128, 'seed': 0, 'dataset_shuffle_seed': 0, 'device': 'cuda', 'save_video': True, 'save_train_video': True, 'use_tb': True, 'use_wandb': True, 'wandb_run_id': '265626', 'wandb_notes': '265626_req_265827', 'eval': True, 'process_contact_features': True, 'obs_type': 'pixels', 'use_color': True, 'use_depth': True, 'use_masks': True, 'mask_list': ['EE_obj_mask'], 'mask_representation': 'channels', 'crop_hw': [144, 144], 'crop_down_offset': 48, 'color_crop_type': None, 'depth_crop_type': None, 'segmask_crop_type': None, 'add_crop_binary_mask': False, 'add_coord_conv_map': False, 'use_context_color': False, 'use_context_depth': False, 'use_context_segmask': False, 'context_color_crop_type': None, 'context_depth_crop_type': None, 'context_segmask_crop_type': None, 'context_add_crop_binary_mask': False, 'context_add_coord_conv_map': False, 'use_contact_map': False, 'use_sdf_maps': False, 'use_normals_maps': False, 'which_objects': 'both', 'max_contact_prob': 0.1, 'max_depth': 2.0, 'grasped_dtc_max_value': 0.2, 'env_dtc_max_value': 0.4, 'grasped_normals_mask_max_dtc_value': 0.2, 'env_normals_mask_max_dtc_value': 0.4, 'clamp_dtc': True, 'dtc_adaptive_normalization': False, 'mask_normals_within_sdf': True, 'adaptive_normals_mask': True, 'learnable_contact_preprocess_params': True, 'contact_model_name': 'local_multitask_outhd64all_home_crop_h144w144d48_mask_ctxtmask_seed_220979_epoch_9', 'contact_estimation_model_ckpt_path': '~/fish_leon/contact_estimation/artifacts/175604_2/checkpoints/epoch=09-val_loss=0.00.ckpt', 'encoder_type': 'small', 'debug_timestamps': False, 'open_loop': False, 'action_trajectories': True, 'stop_after_action': False, 'interpolation_frequency': 25, 'policy_frequency': 5, 'wait_for_new_camera_frames': True, 'baseline': False, 'train_demo_idxs_list_or_num': -1, 'log_train_every_steps': 25, 'name_of_expert_demo': '64_240x320_hbm_twodim_fps_fix_contact_fix_recovery_fix_annotated_start_idx_5hz_zstd7_EE_pxl_coords_expert_demos_imp_act', 'expert_dataset_dirpath': '/home/leonmkim/fish_leon/FISH/expert_demos/frankagym/FrankaInsertion-v1/64_240x320_hbm_twodim_fps_fix_contact_fix_recovery_fix_annotated_start_idx_5hz_zstd7_EE_pxl_coords_expert_demos_imp_act', 'store_dataset_in_memory': False, 'expert_dataset': '/home/leonmkim/fish_leon/FISH/expert_demos/frankagym/FrankaInsertion-v1/64_240x320_hbm_twodim_fps_fix_contact_fix_recovery_fix_annotated_start_idx_5hz_zstd7_EE_pxl_coords_expert_demos_imp_act/demos.zarr', 'action_key': 'action_trajectory_25hz', 'semantic_demo_grouping_name': 'semantic_demo_grouping.yaml', 'semantic_demo_grouping': '/home/leonmkim/fish_leon/FISH/expert_demos/frankagym/FrankaInsertion-v1/64_240x320_hbm_twodim_fps_fix_contact_fix_recovery_fix_annotated_start_idx_5hz_zstd7_EE_pxl_coords_expert_demos_imp_act/semantic_demo_grouping.yaml', 'include_groups_list': ['hbm_twodim_left_to_right'], 'expert_dataset_config': '/home/leonmkim/fish_leon/FISH/expert_demos/frankagym/FrankaInsertion-v1/64_240x320_hbm_twodim_fps_fix_contact_fix_recovery_fix_annotated_start_idx_5hz_zstd7_EE_pxl_coords_expert_demos_imp_act/demo_config.yaml', 'name_of_valid_demo': '64_240x320_hbm_twodim_fps_fix_contact_fix_recovery_fix_annotated_start_idx_5hz_zstd7_EE_pxl_coords_expert_demos_imp_act', 'valid_dataset_dir': '/home/leonmkim/fish_leon/FISH/expert_demos/frankagym/FrankaInsertion-v1/64_240x320_hbm_twodim_fps_fix_contact_fix_recovery_fix_annotated_start_idx_5hz_zstd7_EE_pxl_coords_expert_demos_imp_act/demos.zarr', 'valid_demo_idxs_list_or_num': None, 'val_num_groups': 0, 'load_bc': True, 'checkpoint_epoch_list': [99, 199, 299, 399, 499, 599, 699, 799, 899, 999, 1249, 1499, 1749, 1999, 2999, 3999, 4999, 5999, 6999, 7999, 8999, 9999], 'snapshot_root_dir': '/mnt/grasp_high_usage/leonmkim/contact_estimation/FISH', 'save_snapshot': True, 'save_last_snapshot': True, 'save_snapshot_when_done': True, 'top_k_checkpoints': 5, 'save_snapshot_link_to_weights_dir': 'deprecated', 'bc_regularize': False, 'bc_weight_type': 'qfilter', 'experiment_dir': './exp_local/frankagym_pixels/FrankaInsertion-v1/265626', 'agent': {'_target_': 'agent.diffusion_policy.DiffusionPolicyAgent', 'name': 'diffusion_policy', 'load_checkpoint': True, 'obs_shape': [5, 240, 320], 'action_shape': [36, 7], 'device': 'cuda', 'n_obs_steps': 1, 'suite_name': 'frankagym', 'obs_type': 'pixels', 'enable_arm': True, 'enable_camera': True, 'use_tb': True, 'desired_image_shape': [13, 180, 240], 'orig_cam_shape': [3, 240, 320], 'config': {'_target_': 'agent.diffusion_policy.DiffusionPolicyAgentConfig', 'compile': False, 'device': 'cuda', 'cam_resize_shape': [13, 180, 240], 'orig_cam_shape': [3, 240, 320], 'policy_cfg': {'_target_': 'lerobot.common.policies.diffusion.configuration_diffusion.DiffusionConfig', 'n_obs_steps': 1, 'horizon': 36, 'n_action_steps': 36, 'output_shapes': {'action': [7]}, 'input_normalization_modes': {'observation.image': 'mean_std', 'observation.state': 'min_max', 'observation.action_history': 'min_max'}, 'output_normalization_modes': {'action': 'min_max'}, 'vision_backbone': 'resnet18', 'pretrained_backbone_weights': None, 'transforms': [{'_target_': 'torchaug.transforms.RandomAffine', 'degrees': [-5, 5], 'translate': [0.05, 0.05], 'batch_transform': True, 'num_chunks': -1, 'batch_inplace': True}, {'_target_': 'torchaug.transforms.RandomColorJitter', 'brightness': 0.3, 'contrast': 0.4, 'saturation': 0.5, 'hue': 0.08, 'batch_transform': True, 'num_chunks': -1, 'batch_inplace': True}], 'use_group_norm': True, 'spatial_softmax_num_keypoints': 32, 'action_history_encoder_config': {'_target_': 'lerobot.common.policies.diffusion.configuration_diffusion.Unet1dEncoderConfig', 'in_channels': 7, 'out_channels': 32, 'history_length': 6, 'kernel_size': 5, 'downsample_kernel_size': 3, 'downsample_stride': 2, 'downsample_padding': 1}, 'down_dims': [256, 512, 1024], 'kernel_size': 5, 'n_groups': 8, 'diffusion_step_embed_dim': 128, 'use_film_scale_modulation': True, 'noise_scheduler_type': 'DDIM', 'beta_schedule': 'squaredcos_cap_v2', 'beta_start': 0.0001, 'beta_end': 0.02, 'prediction_type': 'epsilon', 'clip_sample': True, 'clip_sample_range': 1.0, 'num_train_timesteps': 50, 'num_inference_steps': 10, 'do_mask_loss_for_padding': False, 'input_shapes': {'observation.image': [13, 180, 240], 'context_observation.image': [13, 180, 240], 'observation.state': [8], 'observation.action_history': [7]}}, 'train_cfg': {'_target_': 'utils.TrainConfig', 'lr': 0.0001, 'lr_scheduler': 'cosine', 'lr_warmup_steps': 500, 'adam_betas': [0.95, 0.999], 'adam_eps': 1e-08, 'adam_weight_decay': 1e-06, 'grad_clip_norm': 10, 'offline_steps': 1000000, 'use_amp': True}, 'observation_cfg': {'_target_': 'agent.encoder.VisualFeatureSet', 'use_depth': True, 'use_color': True, 'mask_input_dict': {'_target_': 'agent.encoder.MaskInputDict', 'enable': True, 'representation': 'channels', 'mask_list': ['EE_obj_mask']}, 'crop_input_config': {'_target_': 'agent.encoder.CropInputConfig', 'color_crop_type': None, 'depth_crop_type': None, 'segmask_crop_type': None, 'crop_hw': [144, 144], 'crop_down_offset': 48, 'add_crop_binary_mask': False, 'add_coord_conv_map': False}, 'context_input_config': {'_target_': 'agent.encoder.ContextInputConfig', 'use_color': False, 'use_depth': False, 'mask_input_dict': {'_target_': 'agent.encoder.MaskInputDict', 'enable': False, 'representation': 'channels', 'mask_list': ['EE_obj_mask']}, 'crop_input_config': {'_target_': 'agent.encoder.CropInputConfig', 'color_crop_type': None, 'depth_crop_type': None, 'segmask_crop_type': None, 'crop_hw': [144, 144], 'crop_down_offset': 48, 'add_crop_binary_mask': False, 'add_coord_conv_map': False}}, 'mask_soft_approx_scheduler_config': {'_target_': 'agent.encoder.MaskSoftApproxSchedulerConfig', 'num_steps': 40000, 'initial_value': 10.0, 'final_value': 1000.0, 'interpolation_scheme': 'cosine'}, 'use_contact_map': False, 'use_sdf_maps': False, 'use_normals_maps': False, 'which_objects': 'both', 'grasped_dtc_max_value': 0.2, 'env_dtc_max_value': 0.4, 'grasped_normals_mask_max_dtc_value': 0.2, 'env_normals_mask_max_dtc_value': 0.4, 'clamp_dtc': True, 'max_contact_prob': 0.1, 'mask_normals_within_sdf': True, 'dtc_adaptive_normalization': False, 'adaptive_normals_mask': True, 'max_depth': 2.0, 'image_shape': [13, 180, 240], 'learnable_contact_preprocess_params': True, 'learning_rate': 0.0001, 'weight_decay': 0.0, 'contact_model_name': 'local_multitask_outhd64all_home_crop_h144w144d48_mask_ctxtmask_seed_220979_epoch_9', 'zero_centered': False}, 'policy_frequency': 5, 'interpolation_frequency': 25}}, 'suite': {'suite': 'frankagym', 'name': 'frankagym', 'frame_stack': 1, 'action_repeat': 1, 'discount': 0.99, 'hidden_dim': 1024, 'num_train_frames': 2010, 'num_seed_frames': 260, 'num_train_epochs': 5000, 'validate_every_epochs': 100, 'validate_diffusion_on_action_loss_every_epochs': 500, 'train_eval_diffusion_on_action_loss_every_epochs': 500, 'check_topk_every_epochs': 10, 'save_snapshot_every_epochs': 5000, 'eval_every_frames': 2000, 'num_eval_episodes': 5, 'save_snapshot': True, 'wait_for_user_to_start_episode': True, 'task_make_fn': {'_target_': 'suite.frankagym.make', 'name': 'FrankaInsertion-v1', 'height': 240, 'width': 320, 'frame_stack': 1, 'action_repeat': 1, 'seed': 0, 'enable_arm': True, 'enable_gripper': True, 'start_with_gripper_open': True, 'enable_camera': True, 'path_to_depth_extrinsics': '~/fish_leon/FISH/cfgs/camera_poses/camera_poses_L515/20240904-122305/color_tf_world.npy', 'contact_estimation_model_ckpt_path': '~/fish_leon/contact_estimation/artifacts/175604_2/checkpoints/epoch=09-val_loss=0.00.ckpt', 'x_limit': [0.2, 0.7], 'y_limit': [-0.4, 0.4], 'z_limit': [-0.05, 0.55], 'device': 'cuda', 'interpolation_frequency': 25, 'policy_frequency': 5, 'debug_timestamps': False, 'stop_after_action': False, 'open_loop': False, 'wait_for_new_camera_frames': True, 'action_key': 'action_trajectory_25hz', 'action_trajectory_horizon': 36, 'action_trajectories': True, 'path_to_zarr_dataset': '/home/leonmkim/fish_leon/FISH/expert_demos/frankagym/FrankaInsertion-v1/64_240x320_hbm_twodim_fps_fix_contact_fix_recovery_fix_annotated_start_idx_5hz_zstd7_EE_pxl_coords_expert_demos_imp_act/demos.zarr', 'observation_cfg': {'_target_': 'agent.encoder.VisualFeatureSet', 'use_depth': True, 'use_color': True, 'mask_input_dict': {'_target_': 'agent.encoder.MaskInputDict', 'enable': True, 'representation': 'channels', 'mask_list': ['EE_obj_mask']}, 'crop_input_config': {'_target_': 'agent.encoder.CropInputConfig', 'color_crop_type': None, 'depth_crop_type': None, 'segmask_crop_type': None, 'crop_hw': [144, 144], 'crop_down_offset': 48, 'add_crop_binary_mask': False, 'add_coord_conv_map': False}, 'context_input_config': {'_target_': 'agent.encoder.ContextInputConfig', 'use_color': False, 'use_depth': False, 'mask_input_dict': {'_target_': 'agent.encoder.MaskInputDict', 'enable': False, 'representation': 'channels', 'mask_list': ['EE_obj_mask']}, 'crop_input_config': {'_target_': 'agent.encoder.CropInputConfig', 'color_crop_type': None, 'depth_crop_type': None, 'segmask_crop_type': None, 'crop_hw': [144, 144], 'crop_down_offset': 48, 'add_crop_binary_mask': False, 'add_coord_conv_map': False}}, 'mask_soft_approx_scheduler_config': {'_target_': 'agent.encoder.MaskSoftApproxSchedulerConfig', 'num_steps': 40000, 'initial_value': 10.0, 'final_value': 1000.0, 'interpolation_scheme': 'cosine'}, 'use_contact_map': False, 'use_sdf_maps': False, 'use_normals_maps': False, 'which_objects': 'both', 'grasped_dtc_max_value': 0.2, 'env_dtc_max_value': 0.4, 'grasped_normals_mask_max_dtc_value': 0.2, 'env_normals_mask_max_dtc_value': 0.4, 'clamp_dtc': True, 'max_contact_prob': 0.1, 'mask_normals_within_sdf': True, 'dtc_adaptive_normalization': False, 'adaptive_normals_mask': True, 'max_depth': 2.0, 'image_shape': [13, 180, 240], 'learnable_contact_preprocess_params': True, 'learning_rate': 0.0001, 'weight_decay': 0.0, 'contact_model_name': 'local_multitask_outhd64all_home_crop_h144w144d48_mask_ctxtmask_seed_220979_epoch_9', 'zero_centered': False}, 'agent_policy_cfg': {'_target_': 'agent.diffusion_policy.DiffusionPolicyAgentConfig', 'compile': False, 'device': 'cuda', 'cam_resize_shape': [13, 180, 240], 'orig_cam_shape': [3, 240, 320], 'policy_cfg': {'_target_': 'lerobot.common.policies.diffusion.configuration_diffusion.DiffusionConfig', 'n_obs_steps': 1, 'horizon': 36, 'n_action_steps': 36, 'output_shapes': {'action': [7]}, 'input_normalization_modes': {'observation.image': 'mean_std', 'observation.state': 'min_max', 'observation.action_history': 'min_max'}, 'output_normalization_modes': {'action': 'min_max'}, 'vision_backbone': 'resnet18', 'pretrained_backbone_weights': None, 'transforms': [{'_target_': 'torchaug.transforms.RandomAffine', 'degrees': [-5, 5], 'translate': [0.05, 0.05], 'batch_transform': True, 'num_chunks': -1, 'batch_inplace': True}, {'_target_': 'torchaug.transforms.RandomColorJitter', 'brightness': 0.3, 'contrast': 0.4, 'saturation': 0.5, 'hue': 0.08, 'batch_transform': True, 'num_chunks': -1, 'batch_inplace': True}], 'use_group_norm': True, 'spatial_softmax_num_keypoints': 32, 'action_history_encoder_config': {'_target_': 'lerobot.common.policies.diffusion.configuration_diffusion.Unet1dEncoderConfig', 'in_channels': 7, 'out_channels': 32, 'history_length': 6, 'kernel_size': 5, 'downsample_kernel_size': 3, 'downsample_stride': 2, 'downsample_padding': 1}, 'down_dims': [256, 512, 1024], 'kernel_size': 5, 'n_groups': 8, 'diffusion_step_embed_dim': 128, 'use_film_scale_modulation': True, 'noise_scheduler_type': 'DDIM', 'beta_schedule': 'squaredcos_cap_v2', 'beta_start': 0.0001, 'beta_end': 0.02, 'prediction_type': 'epsilon', 'clip_sample': True, 'clip_sample_range': 1.0, 'num_train_timesteps': 50, 'num_inference_steps': 10, 'do_mask_loss_for_padding': False, 'input_shapes': {'observation.image': [13, 180, 240], 'context_observation.image': [13, 180, 240], 'observation.state': [8], 'observation.action_history': [7]}}, 'train_cfg': {'_target_': 'utils.TrainConfig', 'lr': 0.0001, 'lr_scheduler': 'cosine', 'lr_warmup_steps': 500, 'adam_betas': [0.95, 0.999], 'adam_eps': 1e-08, 'adam_weight_decay': 1e-06, 'grad_clip_norm': 10, 'offline_steps': 1000000, 'use_amp': True}, 'observation_cfg': {'_target_': 'agent.encoder.VisualFeatureSet', 'use_depth': True, 'use_color': True, 'mask_input_dict': {'_target_': 'agent.encoder.MaskInputDict', 'enable': True, 'representation': 'channels', 'mask_list': ['EE_obj_mask']}, 'crop_input_config': {'_target_': 'agent.encoder.CropInputConfig', 'color_crop_type': None, 'depth_crop_type': None, 'segmask_crop_type': None, 'crop_hw': [144, 144], 'crop_down_offset': 48, 'add_crop_binary_mask': False, 'add_coord_conv_map': False}, 'context_input_config': {'_target_': 'agent.encoder.ContextInputConfig', 'use_color': False, 'use_depth': False, 'mask_input_dict': {'_target_': 'agent.encoder.MaskInputDict', 'enable': False, 'representation': 'channels', 'mask_list': ['EE_obj_mask']}, 'crop_input_config': {'_target_': 'agent.encoder.CropInputConfig', 'color_crop_type': None, 'depth_crop_type': None, 'segmask_crop_type': None, 'crop_hw': [144, 144], 'crop_down_offset': 48, 'add_crop_binary_mask': False, 'add_coord_conv_map': False}}, 'mask_soft_approx_scheduler_config': {'_target_': 'agent.encoder.MaskSoftApproxSchedulerConfig', 'num_steps': 40000, 'initial_value': 10.0, 'final_value': 1000.0, 'interpolation_scheme': 'cosine'}, 'use_contact_map': False, 'use_sdf_maps': False, 'use_normals_maps': False, 'which_objects': 'both', 'grasped_dtc_max_value': 0.2, 'env_dtc_max_value': 0.4, 'grasped_normals_mask_max_dtc_value': 0.2, 'env_normals_mask_max_dtc_value': 0.4, 'clamp_dtc': True, 'max_contact_prob': 0.1, 'mask_normals_within_sdf': True, 'dtc_adaptive_normalization': False, 'adaptive_normals_mask': True, 'max_depth': 2.0, 'image_shape': [13, 180, 240], 'learnable_contact_preprocess_params': True, 'learning_rate': 0.0001, 'weight_decay': 0.0, 'contact_model_name': 'local_multitask_outhd64all_home_crop_h144w144d48_mask_ctxtmask_seed_220979_epoch_9', 'zero_centered': False}, 'policy_frequency': 5, 'interpolation_frequency': 25}, 'true_action_history': True}}, 'num_train_frames_bc': 50000, 'num_train_frames_drq': 1100000, 'stddev_schedule_drq': 'linear(1.0,0.1,100000)', 'task_name': 'FrankaInsertion-v1', 'num_train_frames_vinn': 25000, 'num_train_frames_diffusion': 1000000, 'num_train_epochs_bc': 5000, 'num_train_epochs_diffusion': 15000, 'validate_every_epochs_bc': 5, 'validate_every_epochs_diffusion': 250, 'validate_diffusion_on_action_loss_every_epochs': 250, 'train_eval_diffusion_on_action_loss_every_epochs': 250, 'check_topk_every_epochs': 5, 'check_topk_every_epochs_diffusion': 250, 'save_snapshot_every_epochs_diffusion': 1500, 'x_limit': [0.2, 0.7], 'y_limit': [-0.4, 0.4], 'z_limit': [-0.05, 0.55], 'home_displacement': [0.55, 0.0, 0.55, 180.0, 0.0, 0.0], 'enable_gripper': True, 'start_with_gripper_open': True, 'offset_mask': [1, 1, 1, 1, 1, 1], 'path_to_depth_extrinsics': '~/fish_leon/FISH/cfgs/camera_poses/camera_poses_L515/20240904-122305/color_tf_world.npy', 'feature_type': '180x240_1_RGB_D_2.0_msk_channels_EE_obj_mask_acthst_hst6_out32_dwnkrnl3_dwnstrd2_dwnpd1', 'save_buffer': True, 'num_eval': 5, 'random_start': False, 'eval_starts': '/home/leonmkim/fish_leon/FISH/eval_starts/frankagym_pixels/FrankaInsertion-v1', 'num_valid_demos': None, 'load_checkpoint': True, 'true_action_history': True, 'checkpoint_epoch': 12000, 'load_residual_weight': False, 'checkpoint_root_dir': '/home/leonmkim/fish_leon/FISH', 'checkpoint_weight_dir': '/home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626', 'residual_weight': '/home/leonmkim/fish_leon/FISH/weights/frankagym_pixels/FrankaInsertion-v1/weight.pt'}
14
+ 2024-12-10 20:51:30,304 INFO MainThread:596892 [wandb_init.py:init():619] starting backend
15
+ 2024-12-10 20:51:30,304 INFO MainThread:596892 [wandb_init.py:init():623] setting up manager
16
+ 2024-12-10 20:51:30,310 INFO MainThread:596892 [backend.py:_multiprocessing_setup():105] multiprocessing start_methods=fork,spawn,forkserver, using: spawn
17
+ 2024-12-10 20:51:30,312 INFO MainThread:596892 [wandb_init.py:init():631] backend started and connected
18
+ 2024-12-10 20:51:30,324 INFO MainThread:596892 [wandb_init.py:init():720] updated telemetry
19
+ 2024-12-10 20:51:30,334 INFO MainThread:596892 [wandb_init.py:init():753] communicating run to backend with 90.0 second timeout
20
+ 2024-12-10 20:51:30,656 INFO MainThread:596892 [wandb_run.py:_on_init():2435] communicating current version
21
+ 2024-12-10 20:51:30,737 INFO MainThread:596892 [wandb_run.py:_on_init():2444] got version response upgrade_message: "wandb version 0.19.0 is available! To upgrade, please run:\n $ pip install wandb --upgrade"
22
+
23
+ 2024-12-10 20:51:30,737 INFO MainThread:596892 [wandb_init.py:init():804] starting run threads in backend
24
+ 2024-12-10 20:51:31,144 INFO MainThread:596892 [wandb_run.py:_console_start():2413] atexit reg
25
+ 2024-12-10 20:51:31,145 INFO MainThread:596892 [wandb_run.py:_redirect():2255] redirect: wrap_raw
26
+ 2024-12-10 20:51:31,145 INFO MainThread:596892 [wandb_run.py:_redirect():2320] Wrapping output streams.
27
+ 2024-12-10 20:51:31,145 INFO MainThread:596892 [wandb_run.py:_redirect():2345] Redirects installed.
28
+ 2024-12-10 20:51:31,147 INFO MainThread:596892 [wandb_init.py:init():847] run started, returning control to user process
29
+ 2024-12-10 20:51:31,147 INFO MainThread:596892 [wandb_run.py:_tensorboard_callback():1544] tensorboard callback: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205123/tb, True
30
+ 2024-12-10 20:51:46,908 INFO MainThread:596892 [wandb_run.py:_config_callback():1382] config_cb None None {'grasped_obj_name': 'hbm', 'left_book_slot': 'twodim'}
31
+ 2024-12-10 20:52:55,379 WARNING MsgRouterThr:596892 [router.py:message_loop():77] message_loop has been closed
205123/wandb/run-20241210_205130-svysfc56/run-svysfc56.wandb ADDED
Binary file (70.1 kB). View file
 
205307/.hydra/config.yaml ADDED
@@ -0,0 +1,349 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ root_dir: /home/${oc.env:USER}/fish_leon
2
+ nstep: 3
3
+ seed: 41
4
+ dataset_shuffle_seed: ${seed}
5
+ device: cuda
6
+ save_video: true
7
+ save_buffer: true
8
+ use_tb: true
9
+ baseline: false
10
+ use_wandb: true
11
+ eval: true
12
+ process_contact_features: ${eval}
13
+ obs_type: pixels
14
+ use_color: true
15
+ use_depth: true
16
+ use_masks: false
17
+ mask_list:
18
+ - EE_obj_mask
19
+ mask_representation: channels
20
+ crop_hw:
21
+ - 144
22
+ - 144
23
+ crop_down_offset: 48
24
+ color_crop_type: null
25
+ depth_crop_type: null
26
+ segmask_crop_type: null
27
+ add_crop_binary_mask: false
28
+ add_coord_conv_map: false
29
+ use_context_color: false
30
+ use_context_depth: false
31
+ use_context_segmask: false
32
+ context_color_crop_type: null
33
+ context_depth_crop_type: null
34
+ context_segmask_crop_type: null
35
+ context_add_crop_binary_mask: false
36
+ context_add_coord_conv_map: false
37
+ use_contact_map: false
38
+ use_sdf_maps: false
39
+ use_normals_maps: false
40
+ which_objects: both
41
+ max_contact_prob: 0.1
42
+ max_depth: 2.0
43
+ grasped_dtc_max_value: 0.105
44
+ env_dtc_max_value: 0.425
45
+ grasped_normals_mask_max_dtc_value: 0.105
46
+ env_normals_mask_max_dtc_value: 0.425
47
+ clamp_dtc: true
48
+ dtc_adaptive_normalization: false
49
+ mask_normals_within_sdf: true
50
+ adaptive_normals_mask: true
51
+ learnable_contact_preprocess_params: false
52
+ contact_model_name: local_multitask_outhd64all_home_crop_h144w144d48_ctxt_seed_183386_epoch_9
53
+ contact_estimation_model_ckpt_path: ~/fish_leon/contact_estimation/artifacts/175604_2/checkpoints/epoch=09-val_loss=0.00.ckpt
54
+ num_eval: 5
55
+ debug_timestamps: false
56
+ open_loop: false
57
+ action_trajectories: true
58
+ stop_after_action: false
59
+ interpolation_frequency: 25
60
+ policy_frequency: 5
61
+ wait_for_new_camera_frames: true
62
+ random_start: false
63
+ eval_starts: ${root_dir}/FISH/eval_starts/${suite.name}_${obs_type}/${task_name}
64
+ train_demo_idxs_list_or_num: null
65
+ num_valid_demos: null
66
+ val_num_groups: 3
67
+ name_of_expert_demo: 192_240x320_multiobj_twodim_recovery_5hz_zstd7_EE_pxl_coords_expert_demos_imp_act
68
+ expert_dataset_dirpath: ${root_dir}/FISH/expert_demos/${suite.name}/${task_name}/${name_of_expert_demo}
69
+ expert_dataset: ${expert_dataset_dirpath}/demos.zarr
70
+ action_key: ${oc.if_else:${action_trajectories}, 'action_trajectory_${interpolation_frequency}hz',
71
+ 'action'}
72
+ semantic_demo_grouping_name: semantic_demo_grouping.yaml
73
+ semantic_demo_grouping: ${expert_dataset_dirpath}/${semantic_demo_grouping_name}
74
+ expert_dataset_config: ${expert_dataset_dirpath}/demo_config.yaml
75
+ bc_regularize: false
76
+ bc_weight_type: qfilter
77
+ load_checkpoint: ${agent.load_checkpoint}
78
+ wandb_run_id: '265626'
79
+ true_action_history: false
80
+ wandb_notes: null
81
+ checkpoint_epoch: 12000
82
+ load_residual_weight: false
83
+ checkpoint_root_dir: /home/${oc.env:USER}/fish_leon/FISH
84
+ checkpoint_weight_dir: ${checkpoint_root_dir}/exp_local/${suite.name}_${obs_type}/${task_name}/${wandb_run_id}
85
+ residual_weight: ${root_dir}/FISH/weights/${suite.name}_${obs_type}/${task_name}/weight.pt
86
+ experiment_dir: ./exp_local/${suite.name}_${obs_type}/${task_name}/${wandb_run_id}
87
+ agent:
88
+ _target_: agent.diffusion_policy.DiffusionPolicyAgent
89
+ name: diffusion_policy
90
+ load_checkpoint: ${eval}
91
+ device: ${device}
92
+ n_obs_steps: ${.config.policy_cfg.n_obs_steps}
93
+ suite_name: ${suite.name}
94
+ obs_type: ${obs_type}
95
+ enable_arm: ${eval}
96
+ enable_camera: ${eval}
97
+ use_tb: ${use_tb}
98
+ desired_image_shape:
99
+ - 13
100
+ - 180
101
+ - 240
102
+ orig_cam_shape:
103
+ - 3
104
+ - 240
105
+ - 320
106
+ config:
107
+ _target_: agent.diffusion_policy.DiffusionPolicyAgentConfig
108
+ compile: false
109
+ device: ${device}
110
+ cam_resize_shape: ${agent.desired_image_shape}
111
+ orig_cam_shape: ${agent.orig_cam_shape}
112
+ policy_frequency: ${policy_frequency}
113
+ interpolation_frequency: ${interpolation_frequency}
114
+ policy_cfg:
115
+ _target_: lerobot.common.policies.diffusion.configuration_diffusion.DiffusionConfig
116
+ n_obs_steps: 1
117
+ horizon: 36
118
+ n_action_steps: ${agent.config.policy_cfg.horizon}
119
+ input_shapes:
120
+ observation.image: ${agent.config.cam_resize_shape}
121
+ context_observation.image: ${agent.config.cam_resize_shape}
122
+ observation.state:
123
+ - 8
124
+ observation.action_history:
125
+ - 7
126
+ output_shapes:
127
+ action:
128
+ - 7
129
+ input_normalization_modes:
130
+ observation.image: mean_std
131
+ observation.state: min_max
132
+ observation.action_history: min_max
133
+ output_normalization_modes:
134
+ action: min_max
135
+ vision_backbone: resnet18
136
+ pretrained_backbone_weights: null
137
+ transforms:
138
+ - _target_: torchaug.transforms.RandomAffine
139
+ degrees:
140
+ - -5
141
+ - 5
142
+ translate:
143
+ - 0.05
144
+ - 0.05
145
+ batch_transform: true
146
+ num_chunks: -1
147
+ batch_inplace: true
148
+ - _target_: torchaug.transforms.RandomColorJitter
149
+ brightness: 0.3
150
+ contrast: 0.4
151
+ saturation: 0.5
152
+ hue: 0.08
153
+ batch_transform: true
154
+ num_chunks: -1
155
+ batch_inplace: true
156
+ use_group_norm: true
157
+ spatial_softmax_num_keypoints: 32
158
+ action_history_encoder_config:
159
+ _target_: lerobot.common.policies.diffusion.configuration_diffusion.Unet1dEncoderConfig
160
+ in_channels: 7
161
+ out_channels: 32
162
+ history_length: ${agent.config.policy_cfg.n_action_steps}
163
+ kernel_size: ${agent.config.policy_cfg.kernel_size}
164
+ downsample_kernel_size: 3
165
+ downsample_stride: 2
166
+ downsample_padding: 1
167
+ down_dims:
168
+ - 256
169
+ - 512
170
+ - 1024
171
+ kernel_size: 5
172
+ n_groups: 8
173
+ diffusion_step_embed_dim: 128
174
+ use_film_scale_modulation: true
175
+ noise_scheduler_type: DDIM
176
+ beta_schedule: squaredcos_cap_v2
177
+ beta_start: 0.0001
178
+ beta_end: 0.02
179
+ prediction_type: epsilon
180
+ clip_sample: true
181
+ clip_sample_range: 1.0
182
+ num_train_timesteps: 50
183
+ num_inference_steps: 10
184
+ do_mask_loss_for_padding: false
185
+ train_cfg:
186
+ _target_: utils.TrainConfig
187
+ lr: 0.0001
188
+ lr_scheduler: cosine
189
+ lr_warmup_steps: 500
190
+ adam_betas:
191
+ - 0.95
192
+ - 0.999
193
+ adam_eps: 1.0e-08
194
+ adam_weight_decay: 1.0e-06
195
+ grad_clip_norm: 10
196
+ offline_steps: ${num_train_frames_diffusion}
197
+ use_amp: true
198
+ observation_cfg:
199
+ _target_: agent.encoder.VisualFeatureSet
200
+ use_depth: ${use_depth}
201
+ use_color: ${use_color}
202
+ mask_input_dict:
203
+ _target_: agent.encoder.MaskInputDict
204
+ enable: ${use_masks}
205
+ representation: ${mask_representation}
206
+ mask_list: ${mask_list}
207
+ crop_input_config:
208
+ _target_: agent.encoder.CropInputConfig
209
+ color_crop_type: ${color_crop_type}
210
+ depth_crop_type: ${depth_crop_type}
211
+ segmask_crop_type: ${segmask_crop_type}
212
+ crop_hw: ${crop_hw}
213
+ crop_down_offset: ${crop_down_offset}
214
+ add_crop_binary_mask: ${add_crop_binary_mask}
215
+ add_coord_conv_map: ${add_coord_conv_map}
216
+ context_input_config:
217
+ _target_: agent.encoder.ContextInputConfig
218
+ use_color: ${use_context_color}
219
+ use_depth: ${use_context_depth}
220
+ mask_input_dict:
221
+ _target_: agent.encoder.MaskInputDict
222
+ enable: ${use_context_segmask}
223
+ representation: ${mask_representation}
224
+ mask_list: ${mask_list}
225
+ crop_input_config:
226
+ _target_: agent.encoder.CropInputConfig
227
+ color_crop_type: ${context_color_crop_type}
228
+ depth_crop_type: ${context_depth_crop_type}
229
+ segmask_crop_type: ${context_segmask_crop_type}
230
+ crop_hw: ${crop_hw}
231
+ crop_down_offset: ${crop_down_offset}
232
+ add_crop_binary_mask: ${context_add_crop_binary_mask}
233
+ add_coord_conv_map: ${context_add_coord_conv_map}
234
+ mask_soft_approx_scheduler_config:
235
+ _target_: agent.encoder.MaskSoftApproxSchedulerConfig
236
+ num_steps: 40000
237
+ initial_value: 10.0
238
+ final_value: 1000.0
239
+ interpolation_scheme: constant
240
+ use_contact_map: ${use_contact_map}
241
+ use_sdf_maps: ${use_sdf_maps}
242
+ use_normals_maps: ${use_normals_maps}
243
+ which_objects: ${which_objects}
244
+ grasped_dtc_max_value: ${grasped_dtc_max_value}
245
+ env_dtc_max_value: ${env_dtc_max_value}
246
+ grasped_normals_mask_max_dtc_value: ${grasped_normals_mask_max_dtc_value}
247
+ env_normals_mask_max_dtc_value: ${env_normals_mask_max_dtc_value}
248
+ clamp_dtc: ${clamp_dtc}
249
+ max_contact_prob: ${max_contact_prob}
250
+ mask_normals_within_sdf: ${mask_normals_within_sdf}
251
+ dtc_adaptive_normalization: ${dtc_adaptive_normalization}
252
+ adaptive_normals_mask: ${adaptive_normals_mask}
253
+ max_depth: ${max_depth}
254
+ image_shape: ${agent.desired_image_shape}
255
+ learnable_contact_preprocess_params: ${learnable_contact_preprocess_params}
256
+ learning_rate: ${agent.config.train_cfg.lr}
257
+ weight_decay: 0.0
258
+ contact_model_name: ${contact_model_name}
259
+ zero_centered: false
260
+ suite:
261
+ suite: frankagym
262
+ name: frankagym
263
+ frame_stack: ${agent.n_obs_steps}
264
+ action_repeat: 1
265
+ discount: 0.99
266
+ hidden_dim: 1024
267
+ num_train_frames: 2010
268
+ num_seed_frames: 260
269
+ num_train_epochs: 5000
270
+ validate_every_epochs: 100
271
+ validate_diffusion_on_action_loss_every_epochs: 500
272
+ train_eval_diffusion_on_action_loss_every_epochs: 500
273
+ check_topk_every_epochs: 10
274
+ save_snapshot_every_epochs: 5000
275
+ eval_every_frames: 2000
276
+ num_eval_episodes: 5
277
+ save_snapshot: true
278
+ wait_for_user_to_start_episode: true
279
+ task_make_fn:
280
+ _target_: suite.frankagym.make
281
+ name: ${task_name}
282
+ height: 240
283
+ width: 320
284
+ frame_stack: ${suite.frame_stack}
285
+ action_repeat: ${suite.action_repeat}
286
+ seed: ${seed}
287
+ enable_arm: ${agent.enable_arm}
288
+ enable_gripper: ${enable_gripper}
289
+ start_with_gripper_open: ${start_with_gripper_open}
290
+ enable_camera: ${agent.enable_camera}
291
+ path_to_depth_extrinsics: ${path_to_depth_extrinsics}
292
+ contact_estimation_model_ckpt_path: ${contact_estimation_model_ckpt_path}
293
+ x_limit: ${x_limit}
294
+ y_limit: ${y_limit}
295
+ z_limit: ${z_limit}
296
+ device: ${device}
297
+ interpolation_frequency: ${interpolation_frequency}
298
+ policy_frequency: ${policy_frequency}
299
+ debug_timestamps: ${debug_timestamps}
300
+ stop_after_action: ${stop_after_action}
301
+ open_loop: ${open_loop}
302
+ wait_for_new_camera_frames: ${wait_for_new_camera_frames}
303
+ action_key: ${action_key}
304
+ action_trajectory_horizon: ${agent.config.policy_cfg.horizon}
305
+ action_trajectories: ${action_trajectories}
306
+ path_to_zarr_dataset: ${expert_dataset}
307
+ agent_policy_cfg: ???
308
+ true_action_history: ${true_action_history}
309
+ num_train_frames_bc: 50000
310
+ num_train_frames_drq: 1100000
311
+ stddev_schedule_drq: linear(1.0,0.1,100000)
312
+ task_name: FrankaInsertion-v1
313
+ num_train_frames_vinn: 25000
314
+ num_train_frames_diffusion: 1000000
315
+ num_train_epochs_bc: 5000
316
+ num_train_epochs_diffusion: 5000
317
+ validate_every_epochs_bc: 5
318
+ validate_every_epochs_diffusion: 25
319
+ validate_diffusion_on_action_loss_every_epochs: 50
320
+ train_eval_diffusion_on_action_loss_every_epochs: 500
321
+ check_topk_every_epochs: 5
322
+ check_topk_every_epochs_diffusion: ${validate_diffusion_on_action_loss_every_epochs}
323
+ save_snapshot_every_epochs_diffusion: 5000
324
+ x_limit:
325
+ - 0.2
326
+ - 0.7
327
+ y_limit:
328
+ - -0.4
329
+ - 0.4
330
+ z_limit:
331
+ - -0.05
332
+ - 0.55
333
+ home_displacement:
334
+ - 0.55
335
+ - 0.0
336
+ - 0.55
337
+ - 180.0
338
+ - 0.0
339
+ - 0.0
340
+ enable_gripper: true
341
+ start_with_gripper_open: true
342
+ offset_mask:
343
+ - 1
344
+ - 1
345
+ - 1
346
+ - 1
347
+ - 1
348
+ - 1
349
+ path_to_depth_extrinsics: ~/fish_leon/FISH/cfgs/camera_poses/camera_poses_L515/20240904-122305/color_tf_world.npy
205307/.hydra/hydra.yaml ADDED
@@ -0,0 +1,169 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ hydra:
2
+ run:
3
+ dir: ${experiment_dir}/${now:%H%M%S}
4
+ sweep:
5
+ dir: ${experiment_dir}/${now:%H%M%S}
6
+ subdir: ${hydra.job.num}
7
+ launcher:
8
+ submitit_folder: ${experiment_dir}/${now:%H%M%S}/.slurm
9
+ timeout_min: 60
10
+ cpus_per_task: null
11
+ gpus_per_node: null
12
+ tasks_per_node: 1
13
+ mem_gb: null
14
+ nodes: 1
15
+ name: ${hydra.job.name}
16
+ stderr_to_stdout: false
17
+ _target_: hydra_plugins.hydra_submitit_launcher.submitit_launcher.LocalLauncher
18
+ sweeper:
19
+ _target_: hydra._internal.core_plugins.basic_sweeper.BasicSweeper
20
+ max_batch_size: null
21
+ params: null
22
+ help:
23
+ app_name: ${hydra.job.name}
24
+ header: '${hydra.help.app_name} is powered by Hydra.
25
+
26
+ '
27
+ footer: 'Powered by Hydra (https://hydra.cc)
28
+
29
+ Use --hydra-help to view Hydra specific help
30
+
31
+ '
32
+ template: '${hydra.help.header}
33
+
34
+ == Configuration groups ==
35
+
36
+ Compose your configuration from those groups (group=option)
37
+
38
+
39
+ $APP_CONFIG_GROUPS
40
+
41
+
42
+ == Config ==
43
+
44
+ Override anything in the config (foo.bar=value)
45
+
46
+
47
+ $CONFIG
48
+
49
+
50
+ ${hydra.help.footer}
51
+
52
+ '
53
+ hydra_help:
54
+ template: 'Hydra (${hydra.runtime.version})
55
+
56
+ See https://hydra.cc for more info.
57
+
58
+
59
+ == Flags ==
60
+
61
+ $FLAGS_HELP
62
+
63
+
64
+ == Configuration groups ==
65
+
66
+ Compose your configuration from those groups (For example, append hydra/job_logging=disabled
67
+ to command line)
68
+
69
+
70
+ $HYDRA_CONFIG_GROUPS
71
+
72
+
73
+ Use ''--cfg hydra'' to Show the Hydra config.
74
+
75
+ '
76
+ hydra_help: ???
77
+ hydra_logging:
78
+ version: 1
79
+ formatters:
80
+ simple:
81
+ format: '[%(asctime)s][HYDRA] %(message)s'
82
+ handlers:
83
+ console:
84
+ class: logging.StreamHandler
85
+ formatter: simple
86
+ stream: ext://sys.stdout
87
+ root:
88
+ level: INFO
89
+ handlers:
90
+ - console
91
+ loggers:
92
+ logging_example:
93
+ level: DEBUG
94
+ disable_existing_loggers: false
95
+ job_logging:
96
+ version: 1
97
+ formatters:
98
+ simple:
99
+ format: '[%(asctime)s][%(name)s][%(levelname)s] - %(message)s'
100
+ handlers:
101
+ console:
102
+ class: logging.StreamHandler
103
+ formatter: simple
104
+ stream: ext://sys.stdout
105
+ file:
106
+ class: logging.FileHandler
107
+ formatter: simple
108
+ filename: ${hydra.runtime.output_dir}/${hydra.job.name}.log
109
+ root:
110
+ level: INFO
111
+ handlers:
112
+ - console
113
+ - file
114
+ disable_existing_loggers: false
115
+ env: {}
116
+ mode: RUN
117
+ searchpath: []
118
+ callbacks: {}
119
+ output_subdir: .hydra
120
+ overrides:
121
+ hydra:
122
+ - hydra.mode=RUN
123
+ task:
124
+ - agent=diffusion
125
+ - suite=frankagym
126
+ - suite/frankagym_task@_global_=insertion
127
+ job:
128
+ name: eval_policy
129
+ chdir: true
130
+ override_dirname: agent=diffusion,suite/frankagym_task@_global_=insertion,suite=frankagym
131
+ id: ???
132
+ num: ???
133
+ config_name: config_eval
134
+ env_set: {}
135
+ env_copy: []
136
+ config:
137
+ override_dirname:
138
+ kv_sep: '='
139
+ item_sep: ','
140
+ exclude_keys: []
141
+ runtime:
142
+ version: 1.3.2
143
+ version_base: '1.1'
144
+ cwd: /home/leonmkim/fish_leon/FISH
145
+ config_sources:
146
+ - path: hydra.conf
147
+ schema: pkg
148
+ provider: hydra
149
+ - path: /home/leonmkim/fish_leon/FISH/cfgs
150
+ schema: file
151
+ provider: main
152
+ - path: ''
153
+ schema: structured
154
+ provider: schema
155
+ output_dir: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205307
156
+ choices:
157
+ suite: frankagym
158
+ suite/frankagym_task@_global_: insertion
159
+ agent: diffusion
160
+ hydra/env: default
161
+ hydra/callbacks: null
162
+ hydra/job_logging: default
163
+ hydra/hydra_logging: default
164
+ hydra/hydra_help: default
165
+ hydra/help: default
166
+ hydra/sweeper: basic
167
+ hydra/launcher: submitit_local
168
+ hydra/output: default
169
+ verbose: false
205307/eval_policy.log ADDED
@@ -0,0 +1,15 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ [2024-12-10 20:53:07,380][py.warnings][WARNING] - /home/leonmkim/fish_leon/FISH/eval_policy.py:428: UserWarning:
2
+ The version_base parameter is not specified.
3
+ Please specify a compatability version level, or None.
4
+ Will assume defaults for version 1.1
5
+ @hydra.main(config_path='cfgs', config_name='config_eval')
6
+
7
+ [2024-12-10 20:53:07,383][py.warnings][WARNING] - /home/leonmkim/fish_leon/FISH/eval_policy.py:365: UserWarning:
8
+ The version_base parameter is not specified.
9
+ Please specify a compatability version level, or None.
10
+ Will assume defaults for version 1.1
11
+ hydra.initialize(
12
+
13
+ [2024-12-10 20:53:08,679][py.warnings][WARNING] - /home/leonmkim/fish_leon/FISH/eval_policy.py:414: FutureWarning: You are using `torch.load` with `weights_only=False` (the current default value), which uses the default pickle module implicitly. It is possible to construct malicious pickle data which will execute arbitrary code during unpickling (See https://github.com/pytorch/pytorch/blob/main/SECURITY.md#untrusted-models for more details). In a future release, the default value for `weights_only` will be flipped to `True`. This limits the functions that could be executed during unpickling. Arbitrary objects will no longer be allowed to be loaded via this mode unless they are explicitly allowlisted by the user via `torch.serialization.add_safe_globals`. We recommend you start setting `weights_only=True` for any use case where you don't have full control of the loaded file. Please open an issue on GitHub for any issues related to this experimental feature.
14
+ payload = torch.load(f)
15
+
205324/.hydra/config.yaml ADDED
@@ -0,0 +1,349 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ root_dir: /home/${oc.env:USER}/fish_leon
2
+ nstep: 3
3
+ seed: 41
4
+ dataset_shuffle_seed: ${seed}
5
+ device: cuda
6
+ save_video: true
7
+ save_buffer: true
8
+ use_tb: true
9
+ baseline: false
10
+ use_wandb: true
11
+ eval: true
12
+ process_contact_features: ${eval}
13
+ obs_type: pixels
14
+ use_color: true
15
+ use_depth: true
16
+ use_masks: false
17
+ mask_list:
18
+ - EE_obj_mask
19
+ mask_representation: channels
20
+ crop_hw:
21
+ - 144
22
+ - 144
23
+ crop_down_offset: 48
24
+ color_crop_type: null
25
+ depth_crop_type: null
26
+ segmask_crop_type: null
27
+ add_crop_binary_mask: false
28
+ add_coord_conv_map: false
29
+ use_context_color: false
30
+ use_context_depth: false
31
+ use_context_segmask: false
32
+ context_color_crop_type: null
33
+ context_depth_crop_type: null
34
+ context_segmask_crop_type: null
35
+ context_add_crop_binary_mask: false
36
+ context_add_coord_conv_map: false
37
+ use_contact_map: false
38
+ use_sdf_maps: false
39
+ use_normals_maps: false
40
+ which_objects: both
41
+ max_contact_prob: 0.1
42
+ max_depth: 2.0
43
+ grasped_dtc_max_value: 0.105
44
+ env_dtc_max_value: 0.425
45
+ grasped_normals_mask_max_dtc_value: 0.105
46
+ env_normals_mask_max_dtc_value: 0.425
47
+ clamp_dtc: true
48
+ dtc_adaptive_normalization: false
49
+ mask_normals_within_sdf: true
50
+ adaptive_normals_mask: true
51
+ learnable_contact_preprocess_params: false
52
+ contact_model_name: local_multitask_outhd64all_home_crop_h144w144d48_ctxt_seed_183386_epoch_9
53
+ contact_estimation_model_ckpt_path: ~/fish_leon/contact_estimation/artifacts/175604_2/checkpoints/epoch=09-val_loss=0.00.ckpt
54
+ num_eval: 5
55
+ debug_timestamps: false
56
+ open_loop: false
57
+ action_trajectories: true
58
+ stop_after_action: false
59
+ interpolation_frequency: 25
60
+ policy_frequency: 5
61
+ wait_for_new_camera_frames: true
62
+ random_start: false
63
+ eval_starts: ${root_dir}/FISH/eval_starts/${suite.name}_${obs_type}/${task_name}
64
+ train_demo_idxs_list_or_num: null
65
+ num_valid_demos: null
66
+ val_num_groups: 3
67
+ name_of_expert_demo: 192_240x320_multiobj_twodim_recovery_5hz_zstd7_EE_pxl_coords_expert_demos_imp_act
68
+ expert_dataset_dirpath: ${root_dir}/FISH/expert_demos/${suite.name}/${task_name}/${name_of_expert_demo}
69
+ expert_dataset: ${expert_dataset_dirpath}/demos.zarr
70
+ action_key: ${oc.if_else:${action_trajectories}, 'action_trajectory_${interpolation_frequency}hz',
71
+ 'action'}
72
+ semantic_demo_grouping_name: semantic_demo_grouping.yaml
73
+ semantic_demo_grouping: ${expert_dataset_dirpath}/${semantic_demo_grouping_name}
74
+ expert_dataset_config: ${expert_dataset_dirpath}/demo_config.yaml
75
+ bc_regularize: false
76
+ bc_weight_type: qfilter
77
+ load_checkpoint: ${agent.load_checkpoint}
78
+ wandb_run_id: '265626'
79
+ true_action_history: false
80
+ wandb_notes: null
81
+ checkpoint_epoch: 12000
82
+ load_residual_weight: false
83
+ checkpoint_root_dir: /home/${oc.env:USER}/fish_leon/FISH
84
+ checkpoint_weight_dir: ${checkpoint_root_dir}/exp_local/${suite.name}_${obs_type}/${task_name}/${wandb_run_id}
85
+ residual_weight: ${root_dir}/FISH/weights/${suite.name}_${obs_type}/${task_name}/weight.pt
86
+ experiment_dir: ./exp_local/${suite.name}_${obs_type}/${task_name}/${wandb_run_id}
87
+ agent:
88
+ _target_: agent.diffusion_policy.DiffusionPolicyAgent
89
+ name: diffusion_policy
90
+ load_checkpoint: ${eval}
91
+ device: ${device}
92
+ n_obs_steps: ${.config.policy_cfg.n_obs_steps}
93
+ suite_name: ${suite.name}
94
+ obs_type: ${obs_type}
95
+ enable_arm: ${eval}
96
+ enable_camera: ${eval}
97
+ use_tb: ${use_tb}
98
+ desired_image_shape:
99
+ - 13
100
+ - 180
101
+ - 240
102
+ orig_cam_shape:
103
+ - 3
104
+ - 240
105
+ - 320
106
+ config:
107
+ _target_: agent.diffusion_policy.DiffusionPolicyAgentConfig
108
+ compile: false
109
+ device: ${device}
110
+ cam_resize_shape: ${agent.desired_image_shape}
111
+ orig_cam_shape: ${agent.orig_cam_shape}
112
+ policy_frequency: ${policy_frequency}
113
+ interpolation_frequency: ${interpolation_frequency}
114
+ policy_cfg:
115
+ _target_: lerobot.common.policies.diffusion.configuration_diffusion.DiffusionConfig
116
+ n_obs_steps: 1
117
+ horizon: 36
118
+ n_action_steps: ${agent.config.policy_cfg.horizon}
119
+ input_shapes:
120
+ observation.image: ${agent.config.cam_resize_shape}
121
+ context_observation.image: ${agent.config.cam_resize_shape}
122
+ observation.state:
123
+ - 8
124
+ observation.action_history:
125
+ - 7
126
+ output_shapes:
127
+ action:
128
+ - 7
129
+ input_normalization_modes:
130
+ observation.image: mean_std
131
+ observation.state: min_max
132
+ observation.action_history: min_max
133
+ output_normalization_modes:
134
+ action: min_max
135
+ vision_backbone: resnet18
136
+ pretrained_backbone_weights: null
137
+ transforms:
138
+ - _target_: torchaug.transforms.RandomAffine
139
+ degrees:
140
+ - -5
141
+ - 5
142
+ translate:
143
+ - 0.05
144
+ - 0.05
145
+ batch_transform: true
146
+ num_chunks: -1
147
+ batch_inplace: true
148
+ - _target_: torchaug.transforms.RandomColorJitter
149
+ brightness: 0.3
150
+ contrast: 0.4
151
+ saturation: 0.5
152
+ hue: 0.08
153
+ batch_transform: true
154
+ num_chunks: -1
155
+ batch_inplace: true
156
+ use_group_norm: true
157
+ spatial_softmax_num_keypoints: 32
158
+ action_history_encoder_config:
159
+ _target_: lerobot.common.policies.diffusion.configuration_diffusion.Unet1dEncoderConfig
160
+ in_channels: 7
161
+ out_channels: 32
162
+ history_length: ${agent.config.policy_cfg.n_action_steps}
163
+ kernel_size: ${agent.config.policy_cfg.kernel_size}
164
+ downsample_kernel_size: 3
165
+ downsample_stride: 2
166
+ downsample_padding: 1
167
+ down_dims:
168
+ - 256
169
+ - 512
170
+ - 1024
171
+ kernel_size: 5
172
+ n_groups: 8
173
+ diffusion_step_embed_dim: 128
174
+ use_film_scale_modulation: true
175
+ noise_scheduler_type: DDIM
176
+ beta_schedule: squaredcos_cap_v2
177
+ beta_start: 0.0001
178
+ beta_end: 0.02
179
+ prediction_type: epsilon
180
+ clip_sample: true
181
+ clip_sample_range: 1.0
182
+ num_train_timesteps: 50
183
+ num_inference_steps: 10
184
+ do_mask_loss_for_padding: false
185
+ train_cfg:
186
+ _target_: utils.TrainConfig
187
+ lr: 0.0001
188
+ lr_scheduler: cosine
189
+ lr_warmup_steps: 500
190
+ adam_betas:
191
+ - 0.95
192
+ - 0.999
193
+ adam_eps: 1.0e-08
194
+ adam_weight_decay: 1.0e-06
195
+ grad_clip_norm: 10
196
+ offline_steps: ${num_train_frames_diffusion}
197
+ use_amp: true
198
+ observation_cfg:
199
+ _target_: agent.encoder.VisualFeatureSet
200
+ use_depth: ${use_depth}
201
+ use_color: ${use_color}
202
+ mask_input_dict:
203
+ _target_: agent.encoder.MaskInputDict
204
+ enable: ${use_masks}
205
+ representation: ${mask_representation}
206
+ mask_list: ${mask_list}
207
+ crop_input_config:
208
+ _target_: agent.encoder.CropInputConfig
209
+ color_crop_type: ${color_crop_type}
210
+ depth_crop_type: ${depth_crop_type}
211
+ segmask_crop_type: ${segmask_crop_type}
212
+ crop_hw: ${crop_hw}
213
+ crop_down_offset: ${crop_down_offset}
214
+ add_crop_binary_mask: ${add_crop_binary_mask}
215
+ add_coord_conv_map: ${add_coord_conv_map}
216
+ context_input_config:
217
+ _target_: agent.encoder.ContextInputConfig
218
+ use_color: ${use_context_color}
219
+ use_depth: ${use_context_depth}
220
+ mask_input_dict:
221
+ _target_: agent.encoder.MaskInputDict
222
+ enable: ${use_context_segmask}
223
+ representation: ${mask_representation}
224
+ mask_list: ${mask_list}
225
+ crop_input_config:
226
+ _target_: agent.encoder.CropInputConfig
227
+ color_crop_type: ${context_color_crop_type}
228
+ depth_crop_type: ${context_depth_crop_type}
229
+ segmask_crop_type: ${context_segmask_crop_type}
230
+ crop_hw: ${crop_hw}
231
+ crop_down_offset: ${crop_down_offset}
232
+ add_crop_binary_mask: ${context_add_crop_binary_mask}
233
+ add_coord_conv_map: ${context_add_coord_conv_map}
234
+ mask_soft_approx_scheduler_config:
235
+ _target_: agent.encoder.MaskSoftApproxSchedulerConfig
236
+ num_steps: 40000
237
+ initial_value: 10.0
238
+ final_value: 1000.0
239
+ interpolation_scheme: constant
240
+ use_contact_map: ${use_contact_map}
241
+ use_sdf_maps: ${use_sdf_maps}
242
+ use_normals_maps: ${use_normals_maps}
243
+ which_objects: ${which_objects}
244
+ grasped_dtc_max_value: ${grasped_dtc_max_value}
245
+ env_dtc_max_value: ${env_dtc_max_value}
246
+ grasped_normals_mask_max_dtc_value: ${grasped_normals_mask_max_dtc_value}
247
+ env_normals_mask_max_dtc_value: ${env_normals_mask_max_dtc_value}
248
+ clamp_dtc: ${clamp_dtc}
249
+ max_contact_prob: ${max_contact_prob}
250
+ mask_normals_within_sdf: ${mask_normals_within_sdf}
251
+ dtc_adaptive_normalization: ${dtc_adaptive_normalization}
252
+ adaptive_normals_mask: ${adaptive_normals_mask}
253
+ max_depth: ${max_depth}
254
+ image_shape: ${agent.desired_image_shape}
255
+ learnable_contact_preprocess_params: ${learnable_contact_preprocess_params}
256
+ learning_rate: ${agent.config.train_cfg.lr}
257
+ weight_decay: 0.0
258
+ contact_model_name: ${contact_model_name}
259
+ zero_centered: false
260
+ suite:
261
+ suite: frankagym
262
+ name: frankagym
263
+ frame_stack: ${agent.n_obs_steps}
264
+ action_repeat: 1
265
+ discount: 0.99
266
+ hidden_dim: 1024
267
+ num_train_frames: 2010
268
+ num_seed_frames: 260
269
+ num_train_epochs: 5000
270
+ validate_every_epochs: 100
271
+ validate_diffusion_on_action_loss_every_epochs: 500
272
+ train_eval_diffusion_on_action_loss_every_epochs: 500
273
+ check_topk_every_epochs: 10
274
+ save_snapshot_every_epochs: 5000
275
+ eval_every_frames: 2000
276
+ num_eval_episodes: 5
277
+ save_snapshot: true
278
+ wait_for_user_to_start_episode: true
279
+ task_make_fn:
280
+ _target_: suite.frankagym.make
281
+ name: ${task_name}
282
+ height: 240
283
+ width: 320
284
+ frame_stack: ${suite.frame_stack}
285
+ action_repeat: ${suite.action_repeat}
286
+ seed: ${seed}
287
+ enable_arm: ${agent.enable_arm}
288
+ enable_gripper: ${enable_gripper}
289
+ start_with_gripper_open: ${start_with_gripper_open}
290
+ enable_camera: ${agent.enable_camera}
291
+ path_to_depth_extrinsics: ${path_to_depth_extrinsics}
292
+ contact_estimation_model_ckpt_path: ${contact_estimation_model_ckpt_path}
293
+ x_limit: ${x_limit}
294
+ y_limit: ${y_limit}
295
+ z_limit: ${z_limit}
296
+ device: ${device}
297
+ interpolation_frequency: ${interpolation_frequency}
298
+ policy_frequency: ${policy_frequency}
299
+ debug_timestamps: ${debug_timestamps}
300
+ stop_after_action: ${stop_after_action}
301
+ open_loop: ${open_loop}
302
+ wait_for_new_camera_frames: ${wait_for_new_camera_frames}
303
+ action_key: ${action_key}
304
+ action_trajectory_horizon: ${agent.config.policy_cfg.horizon}
305
+ action_trajectories: ${action_trajectories}
306
+ path_to_zarr_dataset: ${expert_dataset}
307
+ agent_policy_cfg: ???
308
+ true_action_history: ${true_action_history}
309
+ num_train_frames_bc: 50000
310
+ num_train_frames_drq: 1100000
311
+ stddev_schedule_drq: linear(1.0,0.1,100000)
312
+ task_name: FrankaInsertion-v1
313
+ num_train_frames_vinn: 25000
314
+ num_train_frames_diffusion: 1000000
315
+ num_train_epochs_bc: 5000
316
+ num_train_epochs_diffusion: 5000
317
+ validate_every_epochs_bc: 5
318
+ validate_every_epochs_diffusion: 25
319
+ validate_diffusion_on_action_loss_every_epochs: 50
320
+ train_eval_diffusion_on_action_loss_every_epochs: 500
321
+ check_topk_every_epochs: 5
322
+ check_topk_every_epochs_diffusion: ${validate_diffusion_on_action_loss_every_epochs}
323
+ save_snapshot_every_epochs_diffusion: 5000
324
+ x_limit:
325
+ - 0.2
326
+ - 0.7
327
+ y_limit:
328
+ - -0.4
329
+ - 0.4
330
+ z_limit:
331
+ - -0.05
332
+ - 0.55
333
+ home_displacement:
334
+ - 0.55
335
+ - 0.0
336
+ - 0.55
337
+ - 180.0
338
+ - 0.0
339
+ - 0.0
340
+ enable_gripper: true
341
+ start_with_gripper_open: true
342
+ offset_mask:
343
+ - 1
344
+ - 1
345
+ - 1
346
+ - 1
347
+ - 1
348
+ - 1
349
+ path_to_depth_extrinsics: ~/fish_leon/FISH/cfgs/camera_poses/camera_poses_L515/20240904-122305/color_tf_world.npy
205324/.hydra/hydra.yaml ADDED
@@ -0,0 +1,169 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ hydra:
2
+ run:
3
+ dir: ${experiment_dir}/${now:%H%M%S}
4
+ sweep:
5
+ dir: ${experiment_dir}/${now:%H%M%S}
6
+ subdir: ${hydra.job.num}
7
+ launcher:
8
+ submitit_folder: ${experiment_dir}/${now:%H%M%S}/.slurm
9
+ timeout_min: 60
10
+ cpus_per_task: null
11
+ gpus_per_node: null
12
+ tasks_per_node: 1
13
+ mem_gb: null
14
+ nodes: 1
15
+ name: ${hydra.job.name}
16
+ stderr_to_stdout: false
17
+ _target_: hydra_plugins.hydra_submitit_launcher.submitit_launcher.LocalLauncher
18
+ sweeper:
19
+ _target_: hydra._internal.core_plugins.basic_sweeper.BasicSweeper
20
+ max_batch_size: null
21
+ params: null
22
+ help:
23
+ app_name: ${hydra.job.name}
24
+ header: '${hydra.help.app_name} is powered by Hydra.
25
+
26
+ '
27
+ footer: 'Powered by Hydra (https://hydra.cc)
28
+
29
+ Use --hydra-help to view Hydra specific help
30
+
31
+ '
32
+ template: '${hydra.help.header}
33
+
34
+ == Configuration groups ==
35
+
36
+ Compose your configuration from those groups (group=option)
37
+
38
+
39
+ $APP_CONFIG_GROUPS
40
+
41
+
42
+ == Config ==
43
+
44
+ Override anything in the config (foo.bar=value)
45
+
46
+
47
+ $CONFIG
48
+
49
+
50
+ ${hydra.help.footer}
51
+
52
+ '
53
+ hydra_help:
54
+ template: 'Hydra (${hydra.runtime.version})
55
+
56
+ See https://hydra.cc for more info.
57
+
58
+
59
+ == Flags ==
60
+
61
+ $FLAGS_HELP
62
+
63
+
64
+ == Configuration groups ==
65
+
66
+ Compose your configuration from those groups (For example, append hydra/job_logging=disabled
67
+ to command line)
68
+
69
+
70
+ $HYDRA_CONFIG_GROUPS
71
+
72
+
73
+ Use ''--cfg hydra'' to Show the Hydra config.
74
+
75
+ '
76
+ hydra_help: ???
77
+ hydra_logging:
78
+ version: 1
79
+ formatters:
80
+ simple:
81
+ format: '[%(asctime)s][HYDRA] %(message)s'
82
+ handlers:
83
+ console:
84
+ class: logging.StreamHandler
85
+ formatter: simple
86
+ stream: ext://sys.stdout
87
+ root:
88
+ level: INFO
89
+ handlers:
90
+ - console
91
+ loggers:
92
+ logging_example:
93
+ level: DEBUG
94
+ disable_existing_loggers: false
95
+ job_logging:
96
+ version: 1
97
+ formatters:
98
+ simple:
99
+ format: '[%(asctime)s][%(name)s][%(levelname)s] - %(message)s'
100
+ handlers:
101
+ console:
102
+ class: logging.StreamHandler
103
+ formatter: simple
104
+ stream: ext://sys.stdout
105
+ file:
106
+ class: logging.FileHandler
107
+ formatter: simple
108
+ filename: ${hydra.runtime.output_dir}/${hydra.job.name}.log
109
+ root:
110
+ level: INFO
111
+ handlers:
112
+ - console
113
+ - file
114
+ disable_existing_loggers: false
115
+ env: {}
116
+ mode: RUN
117
+ searchpath: []
118
+ callbacks: {}
119
+ output_subdir: .hydra
120
+ overrides:
121
+ hydra:
122
+ - hydra.mode=RUN
123
+ task:
124
+ - agent=diffusion
125
+ - suite=frankagym
126
+ - suite/frankagym_task@_global_=insertion
127
+ job:
128
+ name: eval_robot
129
+ chdir: true
130
+ override_dirname: agent=diffusion,suite/frankagym_task@_global_=insertion,suite=frankagym
131
+ id: ???
132
+ num: ???
133
+ config_name: config_eval
134
+ env_set: {}
135
+ env_copy: []
136
+ config:
137
+ override_dirname:
138
+ kv_sep: '='
139
+ item_sep: ','
140
+ exclude_keys: []
141
+ runtime:
142
+ version: 1.3.2
143
+ version_base: '1.1'
144
+ cwd: /home/leonmkim/fish_leon/FISH
145
+ config_sources:
146
+ - path: hydra.conf
147
+ schema: pkg
148
+ provider: hydra
149
+ - path: /home/leonmkim/fish_leon/FISH/cfgs
150
+ schema: file
151
+ provider: main
152
+ - path: ''
153
+ schema: structured
154
+ provider: schema
155
+ output_dir: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205324
156
+ choices:
157
+ suite: frankagym
158
+ suite/frankagym_task@_global_: insertion
159
+ agent: diffusion
160
+ hydra/env: default
161
+ hydra/callbacks: null
162
+ hydra/job_logging: default
163
+ hydra/hydra_logging: default
164
+ hydra/hydra_help: default
165
+ hydra/help: default
166
+ hydra/sweeper: basic
167
+ hydra/launcher: submitit_local
168
+ hydra/output: default
169
+ verbose: false
205324/.hydra/overrides.yaml ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ - agent=diffusion
2
+ - suite=frankagym
3
+ - suite/frankagym_task@_global_=insertion
205324/episode_rosbags/aligned_depth_to_color_K.npy ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4a962c703d20da282f5e009d432dff51df4ebd22f3386699b6754ea9cfc06a55
3
+ size 200
205324/episode_rosbags/cam_tf_world.npy ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:29313ba240dd65ebbc79056d5582a97c9586cf2a1d4a1e0db13b87b49152cc9e
3
+ size 256
205324/episode_rosbags/color_K.npy ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4a962c703d20da282f5e009d432dff51df4ebd22f3386699b6754ea9cfc06a55
3
+ size 200
205324/episode_rosbags/depth_K.npy ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:cc601d2fecd31c5513c76a88b5d4d8059adc1f92dad682e2d755d89a66d8fdf7
3
+ size 200
205324/eval_robot.log ADDED
@@ -0,0 +1,12 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ [2024-12-10 20:53:24,743][py.warnings][WARNING] - /home/leonmkim/fish_leon/FISH/eval_robot.py:503: UserWarning:
2
+ The version_base parameter is not specified.
3
+ Please specify a compatability version level, or None.
4
+ Will assume defaults for version 1.1
5
+ @hydra.main(config_path='cfgs', config_name='config_eval')
6
+
7
+ [2024-12-10 20:53:24,746][py.warnings][WARNING] - /home/leonmkim/fish_leon/FISH/eval_robot.py:439: UserWarning:
8
+ The version_base parameter is not specified.
9
+ Please specify a compatability version level, or None.
10
+ Will assume defaults for version 1.1
11
+ hydra.initialize(
12
+
205324/wandb/debug-internal.log ADDED
@@ -0,0 +1,327 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ 2024-12-10 20:53:30,850 INFO StreamThr :598006 [internal.py:wandb_internal():85] W&B internal server running at pid: 598006, started at: 2024-12-10 20:53:30.850039
2
+ 2024-12-10 20:53:30,851 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: status
3
+ 2024-12-10 20:53:30,863 INFO WriterThread:598006 [datastore.py:open_for_write():87] open: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205324/wandb/run-20241210_205330-52nddla7/run-52nddla7.wandb
4
+ 2024-12-10 20:53:30,863 DEBUG SenderThread:598006 [sender.py:send():379] send: header
5
+ 2024-12-10 20:53:30,875 DEBUG SenderThread:598006 [sender.py:send():379] send: run
6
+ 2024-12-10 20:53:31,147 INFO SenderThread:598006 [dir_watcher.py:__init__():211] watching files in: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205324/wandb/run-20241210_205330-52nddla7/files
7
+ 2024-12-10 20:53:31,147 INFO SenderThread:598006 [sender.py:_start_run_threads():1188] run started: 52nddla7 with start time 1733882010.850819
8
+ 2024-12-10 20:53:31,161 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: check_version
9
+ 2024-12-10 20:53:31,161 DEBUG SenderThread:598006 [sender.py:send_request():406] send_request: check_version
10
+ 2024-12-10 20:53:31,206 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: run_start
11
+ 2024-12-10 20:53:31,211 DEBUG HandlerThread:598006 [system_info.py:__init__():26] System info init
12
+ 2024-12-10 20:53:31,211 DEBUG HandlerThread:598006 [system_info.py:__init__():41] System info init done
13
+ 2024-12-10 20:53:31,211 INFO HandlerThread:598006 [system_monitor.py:start():194] Starting system monitor
14
+ 2024-12-10 20:53:31,211 INFO SystemMonitor:598006 [system_monitor.py:_start():158] Starting system asset monitoring threads
15
+ 2024-12-10 20:53:31,212 INFO HandlerThread:598006 [system_monitor.py:probe():214] Collecting system info
16
+ 2024-12-10 20:53:31,212 INFO SystemMonitor:598006 [interfaces.py:start():188] Started cpu monitoring
17
+ 2024-12-10 20:53:31,212 INFO SystemMonitor:598006 [interfaces.py:start():188] Started disk monitoring
18
+ 2024-12-10 20:53:31,213 INFO SystemMonitor:598006 [interfaces.py:start():188] Started gpu monitoring
19
+ 2024-12-10 20:53:31,213 INFO SystemMonitor:598006 [interfaces.py:start():188] Started memory monitoring
20
+ 2024-12-10 20:53:31,214 INFO SystemMonitor:598006 [interfaces.py:start():188] Started network monitoring
21
+ 2024-12-10 20:53:31,224 DEBUG HandlerThread:598006 [system_info.py:probe():152] Probing system
22
+ 2024-12-10 20:53:31,226 DEBUG HandlerThread:598006 [system_info.py:_probe_git():137] Probing git
23
+ 2024-12-10 20:53:31,231 DEBUG HandlerThread:598006 [system_info.py:_probe_git():145] Probing git done
24
+ 2024-12-10 20:53:31,231 DEBUG HandlerThread:598006 [system_info.py:probe():200] Probing system done
25
+ 2024-12-10 20:53:31,231 DEBUG HandlerThread:598006 [system_monitor.py:probe():223] {'os': 'Linux-5.15.0-125-generic-x86_64-with-glibc2.31', 'python': '3.10.14', 'heartbeatAt': '2024-12-11T01:53:31.224621', 'startedAt': '2024-12-11T01:53:30.842354', 'docker': None, 'cuda': None, 'args': ('agent=diffusion', 'suite=frankagym', 'suite/frankagym_task@_global_=insertion'), 'state': 'running', 'program': '/home/leonmkim/fish_leon/FISH/eval_robot.py', 'codePathLocal': None, 'codePath': 'FISH/eval_robot.py', 'git': {'remote': 'https://github.com/leonmkim/fish_leon.git', 'commit': '4f7c0421eb84a7caf6bf5670f68777523362676f'}, 'email': 'leonmkim@seas.upenn.edu', 'root': '/home/leonmkim/fish_leon', 'host': 'leonmkim-ROG-Strix-G15CS-G15CS', 'username': 'leonmkim', 'executable': '/home/leonmkim/.pyenv/versions/lerobot/bin/python', 'cpu_count': 8, 'cpu_count_logical': 8, 'cpu_freq': {'current': 4312.5025, 'min': 800.0, 'max': 4700.0}, 'cpu_freq_per_core': [{'current': 4500.019, 'min': 800.0, 'max': 4700.0}, {'current': 4500.002, 'min': 800.0, 'max': 4700.0}, {'current': 3000.0, 'min': 800.0, 'max': 4700.0}, {'current': 4500.007, 'min': 800.0, 'max': 4700.0}, {'current': 4499.999, 'min': 800.0, 'max': 4700.0}, {'current': 3000.0, 'min': 800.0, 'max': 4700.0}, {'current': 3000.0, 'min': 800.0, 'max': 4700.0}, {'current': 4499.943, 'min': 800.0, 'max': 4700.0}], 'disk': {'/': {'total': 915.3232879638672, 'used': 365.2335090637207}}, 'gpu': 'NVIDIA GeForce RTX 2070 SUPER', 'gpu_count': 1, 'gpu_devices': [{'name': 'NVIDIA GeForce RTX 2070 SUPER', 'memory_total': 8589934592}], 'memory': {'total': 62.71596908569336}}
26
+ 2024-12-10 20:53:31,231 INFO HandlerThread:598006 [system_monitor.py:probe():224] Finished collecting system info
27
+ 2024-12-10 20:53:31,231 INFO HandlerThread:598006 [system_monitor.py:probe():227] Publishing system info
28
+ 2024-12-10 20:53:31,231 DEBUG HandlerThread:598006 [system_info.py:_save_code():44] Saving code
29
+ 2024-12-10 20:53:31,233 DEBUG HandlerThread:598006 [system_info.py:_save_code():67] Saving code done
30
+ 2024-12-10 20:53:31,233 DEBUG HandlerThread:598006 [system_info.py:_save_patches():84] Saving git patches
31
+ 2024-12-10 20:53:31,305 DEBUG HandlerThread:598006 [system_info.py:_save_patches():126] Saving git patches done
32
+ 2024-12-10 20:53:31,306 INFO HandlerThread:598006 [system_monitor.py:probe():229] Finished publishing system info
33
+ 2024-12-10 20:53:31,309 DEBUG SenderThread:598006 [sender.py:send():379] send: files
34
+ 2024-12-10 20:53:31,310 INFO SenderThread:598006 [sender.py:_save_file():1454] saving file wandb-metadata.json with policy now
35
+ 2024-12-10 20:53:31,310 INFO SenderThread:598006 [sender.py:_save_file():1454] saving file code/FISH/eval_robot.py with policy now
36
+ 2024-12-10 20:53:31,310 INFO SenderThread:598006 [sender.py:_save_file():1454] saving file diff.patch with policy now
37
+ 2024-12-10 20:53:31,570 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: python_packages
38
+ 2024-12-10 20:53:31,570 DEBUG SenderThread:598006 [sender.py:send_request():406] send_request: python_packages
39
+ 2024-12-10 20:53:31,570 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: stop_status
40
+ 2024-12-10 20:53:31,571 INFO wandb-upload_0:598006 [upload_job.py:push():130] Uploaded file /tmp/tmptbytljn2wandb/oqlixunj-wandb-metadata.json
41
+ 2024-12-10 20:53:31,571 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
42
+ 2024-12-10 20:53:31,572 DEBUG SenderThread:598006 [sender.py:send_request():406] send_request: stop_status
43
+ 2024-12-10 20:53:31,573 INFO HandlerThread:598006 [handler.py:handle_tbrecord():783] handling tbrecord: tbrecord {
44
+ log_dir: "/home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205324/tb"
45
+ save: true
46
+ }
47
+ control {
48
+ relay_id: "5d813b8bb5a547c8896a7e03615dca84"
49
+ }
50
+ _info {
51
+ stream_id: "52nddla7"
52
+ }
53
+
54
+ 2024-12-10 20:53:31,574 DEBUG HandlerThread:598006 [config_util.py:dict_from_config_file():70] no default config file found in config-defaults.yaml
55
+ 2024-12-10 20:53:31,605 INFO wandb-upload_1:598006 [upload_job.py:push():130] Uploaded file /tmp/tmptbytljn2wandb/6yv6eek3-code/FISH/eval_robot.py
56
+ 2024-12-10 20:53:31,648 INFO wandb-upload_2:598006 [upload_job.py:push():130] Uploaded file /tmp/tmptbytljn2wandb/w33l6u66-diff.patch
57
+ 2024-12-10 20:53:31,702 DEBUG SenderThread:598006 [sender.py:send():379] send: telemetry
58
+ 2024-12-10 20:53:31,702 DEBUG SenderThread:598006 [sender.py:send():379] send: tbrecord
59
+ 2024-12-10 20:53:31,703 DEBUG SenderThread:598006 [sender.py:send():379] send: files
60
+ 2024-12-10 20:53:31,703 INFO SenderThread:598006 [sender.py:_save_file():1454] saving file events.out.tfevents.1733882011.leonmkim-ROG-Strix-G15CS-G15CS.597884.0 with policy live
61
+ 2024-12-10 20:53:31,907 INFO wandb-upload_0:598006 [upload_job.py:push():130] Uploaded file /tmp/tmptbytljn2wandb/uat8jiac-events.out.tfevents.1733882011.leonmkim-ROG-Strix-G15CS-G15CS.597884.0
62
+ 2024-12-10 20:53:32,149 INFO Thread-12 :598006 [dir_watcher.py:_on_file_created():271] file/dir created: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205324/wandb/run-20241210_205330-52nddla7/files/events.out.tfevents.1733882011.leonmkim-ROG-Strix-G15CS-G15CS.597884.0
63
+ 2024-12-10 20:53:32,149 INFO Thread-12 :598006 [dir_watcher.py:_on_file_created():271] file/dir created: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205324/wandb/run-20241210_205330-52nddla7/files/code/FISH/eval_robot.py
64
+ 2024-12-10 20:53:32,149 INFO Thread-12 :598006 [dir_watcher.py:_on_file_created():271] file/dir created: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205324/wandb/run-20241210_205330-52nddla7/files/wandb-metadata.json
65
+ 2024-12-10 20:53:32,149 INFO Thread-12 :598006 [dir_watcher.py:_on_file_created():271] file/dir created: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205324/wandb/run-20241210_205330-52nddla7/files/output.log
66
+ 2024-12-10 20:53:32,149 INFO Thread-12 :598006 [dir_watcher.py:_on_file_created():271] file/dir created: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205324/wandb/run-20241210_205330-52nddla7/files/requirements.txt
67
+ 2024-12-10 20:53:32,149 INFO Thread-12 :598006 [dir_watcher.py:_on_file_created():271] file/dir created: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205324/wandb/run-20241210_205330-52nddla7/files/diff.patch
68
+ 2024-12-10 20:53:32,149 INFO Thread-12 :598006 [dir_watcher.py:_on_file_created():271] file/dir created: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205324/wandb/run-20241210_205330-52nddla7/files/code/FISH
69
+ 2024-12-10 20:53:32,149 INFO Thread-12 :598006 [dir_watcher.py:_on_file_created():271] file/dir created: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205324/wandb/run-20241210_205330-52nddla7/files/code
70
+ 2024-12-10 20:53:32,571 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
71
+ 2024-12-10 20:53:33,571 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
72
+ 2024-12-10 20:53:34,148 INFO Thread-12 :598006 [dir_watcher.py:_on_file_modified():288] file/dir modified: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205324/wandb/run-20241210_205330-52nddla7/files/output.log
73
+ 2024-12-10 20:53:34,571 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
74
+ 2024-12-10 20:53:35,547 DEBUG SenderThread:598006 [sender.py:send():379] send: config
75
+ 2024-12-10 20:53:35,590 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
76
+ 2024-12-10 20:53:36,571 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: status_report
77
+ 2024-12-10 20:53:36,572 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
78
+ 2024-12-10 20:53:37,572 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
79
+ 2024-12-10 20:53:38,149 INFO Thread-12 :598006 [dir_watcher.py:_on_file_modified():288] file/dir modified: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205324/wandb/run-20241210_205330-52nddla7/files/output.log
80
+ 2024-12-10 20:53:38,572 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
81
+ 2024-12-10 20:53:39,572 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
82
+ 2024-12-10 20:53:40,150 INFO Thread-12 :598006 [dir_watcher.py:_on_file_modified():288] file/dir modified: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205324/wandb/run-20241210_205330-52nddla7/files/output.log
83
+ 2024-12-10 20:53:40,572 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
84
+ 2024-12-10 20:53:41,572 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
85
+ 2024-12-10 20:53:42,082 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: status_report
86
+ 2024-12-10 20:53:42,150 INFO Thread-12 :598006 [dir_watcher.py:_on_file_modified():288] file/dir modified: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205324/wandb/run-20241210_205330-52nddla7/files/output.log
87
+ 2024-12-10 20:53:42,572 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
88
+ 2024-12-10 20:53:43,572 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
89
+ 2024-12-10 20:53:44,151 INFO Thread-12 :598006 [dir_watcher.py:_on_file_modified():288] file/dir modified: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205324/wandb/run-20241210_205330-52nddla7/files/output.log
90
+ 2024-12-10 20:53:44,573 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
91
+ 2024-12-10 20:53:45,573 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
92
+ 2024-12-10 20:53:46,151 INFO Thread-12 :598006 [dir_watcher.py:_on_file_modified():288] file/dir modified: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205324/wandb/run-20241210_205330-52nddla7/files/output.log
93
+ 2024-12-10 20:53:46,570 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: stop_status
94
+ 2024-12-10 20:53:46,570 DEBUG SenderThread:598006 [sender.py:send_request():406] send_request: stop_status
95
+ 2024-12-10 20:53:46,614 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
96
+ 2024-12-10 20:53:47,573 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
97
+ 2024-12-10 20:53:47,640 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: status_report
98
+ 2024-12-10 20:53:48,573 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
99
+ 2024-12-10 20:53:49,573 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
100
+ 2024-12-10 20:53:50,573 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
101
+ 2024-12-10 20:53:51,574 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
102
+ 2024-12-10 20:53:52,574 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
103
+ 2024-12-10 20:53:53,571 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: status_report
104
+ 2024-12-10 20:53:53,574 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
105
+ 2024-12-10 20:53:54,574 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
106
+ 2024-12-10 20:53:55,574 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
107
+ 2024-12-10 20:53:56,574 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
108
+ 2024-12-10 20:53:57,574 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
109
+ 2024-12-10 20:53:58,571 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: status_report
110
+ 2024-12-10 20:53:58,574 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
111
+ 2024-12-10 20:53:59,575 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
112
+ 2024-12-10 20:54:00,575 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
113
+ 2024-12-10 20:54:01,570 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: stop_status
114
+ 2024-12-10 20:54:01,570 DEBUG SenderThread:598006 [sender.py:send_request():406] send_request: stop_status
115
+ 2024-12-10 20:54:01,613 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
116
+ 2024-12-10 20:54:02,575 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
117
+ 2024-12-10 20:54:03,575 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
118
+ 2024-12-10 20:54:03,722 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: status_report
119
+ 2024-12-10 20:54:04,156 INFO Thread-12 :598006 [dir_watcher.py:_on_file_modified():288] file/dir modified: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205324/wandb/run-20241210_205330-52nddla7/files/config.yaml
120
+ 2024-12-10 20:54:04,575 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
121
+ 2024-12-10 20:54:05,575 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
122
+ 2024-12-10 20:54:06,575 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
123
+ 2024-12-10 20:54:07,576 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
124
+ 2024-12-10 20:54:08,576 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
125
+ 2024-12-10 20:54:09,550 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: status_report
126
+ 2024-12-10 20:54:09,576 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
127
+ 2024-12-10 20:54:10,158 INFO Thread-12 :598006 [dir_watcher.py:_on_file_modified():288] file/dir modified: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205324/wandb/run-20241210_205330-52nddla7/files/output.log
128
+ 2024-12-10 20:54:10,576 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
129
+ 2024-12-10 20:54:11,576 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
130
+ 2024-12-10 20:54:12,159 INFO Thread-12 :598006 [dir_watcher.py:_on_file_modified():288] file/dir modified: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205324/wandb/run-20241210_205330-52nddla7/files/output.log
131
+ 2024-12-10 20:54:12,577 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
132
+ 2024-12-10 20:54:13,577 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
133
+ 2024-12-10 20:54:14,159 INFO Thread-12 :598006 [dir_watcher.py:_on_file_modified():288] file/dir modified: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205324/wandb/run-20241210_205330-52nddla7/files/output.log
134
+ 2024-12-10 20:54:14,567 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: status_report
135
+ 2024-12-10 20:54:14,577 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
136
+ 2024-12-10 20:54:15,160 INFO Thread-12 :598006 [dir_watcher.py:_on_file_modified():288] file/dir modified: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205324/wandb/run-20241210_205330-52nddla7/files/output.log
137
+ 2024-12-10 20:54:15,577 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
138
+ 2024-12-10 20:54:16,570 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: stop_status
139
+ 2024-12-10 20:54:16,570 DEBUG SenderThread:598006 [sender.py:send_request():406] send_request: stop_status
140
+ 2024-12-10 20:54:16,577 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
141
+ 2024-12-10 20:54:17,160 INFO Thread-12 :598006 [dir_watcher.py:_on_file_modified():288] file/dir modified: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205324/wandb/run-20241210_205330-52nddla7/files/output.log
142
+ 2024-12-10 20:54:17,577 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
143
+ 2024-12-10 20:54:18,579 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
144
+ 2024-12-10 20:54:19,161 INFO Thread-12 :598006 [dir_watcher.py:_on_file_modified():288] file/dir modified: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205324/wandb/run-20241210_205330-52nddla7/files/output.log
145
+ 2024-12-10 20:54:19,567 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: status_report
146
+ 2024-12-10 20:54:19,577 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
147
+ 2024-12-10 20:54:20,577 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
148
+ 2024-12-10 20:54:21,161 INFO Thread-12 :598006 [dir_watcher.py:_on_file_modified():288] file/dir modified: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205324/wandb/run-20241210_205330-52nddla7/files/output.log
149
+ 2024-12-10 20:54:21,577 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
150
+ 2024-12-10 20:54:22,578 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
151
+ 2024-12-10 20:54:23,162 INFO Thread-12 :598006 [dir_watcher.py:_on_file_modified():288] file/dir modified: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205324/wandb/run-20241210_205330-52nddla7/files/output.log
152
+ 2024-12-10 20:54:23,578 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
153
+ 2024-12-10 20:54:24,578 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
154
+ 2024-12-10 20:54:24,607 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: status_report
155
+ 2024-12-10 20:54:25,162 INFO Thread-12 :598006 [dir_watcher.py:_on_file_modified():288] file/dir modified: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205324/wandb/run-20241210_205330-52nddla7/files/output.log
156
+ 2024-12-10 20:54:25,578 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
157
+ 2024-12-10 20:54:26,578 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
158
+ 2024-12-10 20:54:27,163 INFO Thread-12 :598006 [dir_watcher.py:_on_file_modified():288] file/dir modified: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205324/wandb/run-20241210_205330-52nddla7/files/output.log
159
+ 2024-12-10 20:54:27,579 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
160
+ 2024-12-10 20:54:28,579 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
161
+ 2024-12-10 20:54:29,164 INFO Thread-12 :598006 [dir_watcher.py:_on_file_modified():288] file/dir modified: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205324/wandb/run-20241210_205330-52nddla7/files/output.log
162
+ 2024-12-10 20:54:29,254 DEBUG SenderThread:598006 [sender.py:send():379] send: files
163
+ 2024-12-10 20:54:29,254 INFO SenderThread:598006 [sender.py:_save_file():1454] saving file media/videos/eval/video_0_ff4edecb6a17ea039fbf.mp4 with policy now
164
+ 2024-12-10 20:54:29,256 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: partial_history
165
+ 2024-12-10 20:54:29,257 DEBUG SenderThread:598006 [sender.py:send():379] send: history
166
+ 2024-12-10 20:54:29,257 DEBUG SenderThread:598006 [sender.py:send_request():406] send_request: summary_record
167
+ 2024-12-10 20:54:29,257 INFO SenderThread:598006 [sender.py:_save_file():1454] saving file wandb-summary.json with policy end
168
+ 2024-12-10 20:54:29,258 DEBUG SenderThread:598006 [sender.py:send():379] send: exit
169
+ 2024-12-10 20:54:29,258 INFO SenderThread:598006 [sender.py:send_exit():586] handling exit code: 0
170
+ 2024-12-10 20:54:29,259 INFO SenderThread:598006 [sender.py:send_exit():588] handling runtime: 58
171
+ 2024-12-10 20:54:29,259 INFO SenderThread:598006 [sender.py:_save_file():1454] saving file wandb-summary.json with policy end
172
+ 2024-12-10 20:54:29,259 INFO SenderThread:598006 [sender.py:send_exit():594] send defer
173
+ 2024-12-10 20:54:29,259 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: defer
174
+ 2024-12-10 20:54:29,259 INFO HandlerThread:598006 [handler.py:handle_request_defer():184] handle defer: 0
175
+ 2024-12-10 20:54:29,259 DEBUG SenderThread:598006 [sender.py:send_request():406] send_request: defer
176
+ 2024-12-10 20:54:29,259 INFO SenderThread:598006 [sender.py:send_request_defer():610] handle sender defer: 0
177
+ 2024-12-10 20:54:29,259 INFO SenderThread:598006 [sender.py:transition_state():614] send defer: 1
178
+ 2024-12-10 20:54:29,259 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: defer
179
+ 2024-12-10 20:54:29,259 INFO HandlerThread:598006 [handler.py:handle_request_defer():184] handle defer: 1
180
+ 2024-12-10 20:54:29,260 DEBUG SenderThread:598006 [sender.py:send_request():406] send_request: defer
181
+ 2024-12-10 20:54:29,260 INFO SenderThread:598006 [sender.py:send_request_defer():610] handle sender defer: 1
182
+ 2024-12-10 20:54:29,260 INFO SenderThread:598006 [sender.py:transition_state():614] send defer: 2
183
+ 2024-12-10 20:54:29,260 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: defer
184
+ 2024-12-10 20:54:29,260 INFO HandlerThread:598006 [handler.py:handle_request_defer():184] handle defer: 2
185
+ 2024-12-10 20:54:29,260 INFO HandlerThread:598006 [system_monitor.py:finish():203] Stopping system monitor
186
+ 2024-12-10 20:54:29,260 DEBUG SystemMonitor:598006 [system_monitor.py:_start():172] Starting system metrics aggregation loop
187
+ 2024-12-10 20:54:29,260 DEBUG SystemMonitor:598006 [system_monitor.py:_start():179] Finished system metrics aggregation loop
188
+ 2024-12-10 20:54:29,260 DEBUG SystemMonitor:598006 [system_monitor.py:_start():183] Publishing last batch of metrics
189
+ 2024-12-10 20:54:29,269 INFO HandlerThread:598006 [interfaces.py:finish():200] Joined cpu monitor
190
+ 2024-12-10 20:54:29,269 INFO HandlerThread:598006 [interfaces.py:finish():200] Joined disk monitor
191
+ 2024-12-10 20:54:29,319 INFO HandlerThread:598006 [interfaces.py:finish():200] Joined gpu monitor
192
+ 2024-12-10 20:54:29,319 INFO HandlerThread:598006 [interfaces.py:finish():200] Joined memory monitor
193
+ 2024-12-10 20:54:29,320 INFO HandlerThread:598006 [interfaces.py:finish():200] Joined network monitor
194
+ 2024-12-10 20:54:29,320 DEBUG SenderThread:598006 [sender.py:send_request():406] send_request: defer
195
+ 2024-12-10 20:54:29,320 INFO SenderThread:598006 [sender.py:send_request_defer():610] handle sender defer: 2
196
+ 2024-12-10 20:54:29,320 INFO SenderThread:598006 [sender.py:transition_state():614] send defer: 3
197
+ 2024-12-10 20:54:29,320 DEBUG SenderThread:598006 [sender.py:send():379] send: stats
198
+ 2024-12-10 20:54:29,320 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: defer
199
+ 2024-12-10 20:54:29,320 INFO HandlerThread:598006 [handler.py:handle_request_defer():184] handle defer: 3
200
+ 2024-12-10 20:54:29,321 DEBUG SenderThread:598006 [sender.py:send_request():406] send_request: defer
201
+ 2024-12-10 20:54:29,321 INFO SenderThread:598006 [sender.py:send_request_defer():610] handle sender defer: 3
202
+ 2024-12-10 20:54:29,321 INFO SenderThread:598006 [sender.py:transition_state():614] send defer: 4
203
+ 2024-12-10 20:54:29,321 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: defer
204
+ 2024-12-10 20:54:29,321 INFO HandlerThread:598006 [handler.py:handle_request_defer():184] handle defer: 4
205
+ 2024-12-10 20:54:29,518 INFO wandb-upload_2:598006 [upload_job.py:push():130] Uploaded file /tmp/tmptbytljn2wandb/0kw42edo-media/videos/eval/video_0_ff4edecb6a17ea039fbf.mp4
206
+ 2024-12-10 20:54:30,164 INFO Thread-12 :598006 [dir_watcher.py:_on_file_modified():288] file/dir modified: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205324/wandb/run-20241210_205330-52nddla7/files/output.log
207
+ 2024-12-10 20:54:30,164 INFO Thread-12 :598006 [dir_watcher.py:_on_file_created():271] file/dir created: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205324/wandb/run-20241210_205330-52nddla7/files/media/videos/eval/video_0_ff4edecb6a17ea039fbf.mp4
208
+ 2024-12-10 20:54:30,164 INFO Thread-12 :598006 [dir_watcher.py:_on_file_created():271] file/dir created: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205324/wandb/run-20241210_205330-52nddla7/files/wandb-summary.json
209
+ 2024-12-10 20:54:30,165 INFO Thread-12 :598006 [dir_watcher.py:_on_file_created():271] file/dir created: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205324/wandb/run-20241210_205330-52nddla7/files/media/videos
210
+ 2024-12-10 20:54:30,165 INFO Thread-12 :598006 [dir_watcher.py:_on_file_created():271] file/dir created: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205324/wandb/run-20241210_205330-52nddla7/files/media/videos/eval
211
+ 2024-12-10 20:54:30,165 INFO Thread-12 :598006 [dir_watcher.py:_on_file_created():271] file/dir created: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205324/wandb/run-20241210_205330-52nddla7/files/media
212
+ 2024-12-10 20:54:30,523 INFO wandb-upload_0:598006 [upload_job.py:push():130] Uploaded file /tmp/tmptbytljn2wandb/ks8s91l3-media/videos/eval/video_0_ff4edecb6a17ea039fbf.mp4
213
+ 2024-12-10 20:54:35,581 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: poll_exit
214
+ 2024-12-10 20:54:35,581 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: status_report
215
+ 2024-12-10 20:54:35,581 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: keepalive
216
+ 2024-12-10 20:54:35,581 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: status_report
217
+ 2024-12-10 20:54:35,581 DEBUG SenderThread:598006 [sender.py:send_request():406] send_request: defer
218
+ 2024-12-10 20:54:35,581 INFO SenderThread:598006 [sender.py:send_request_defer():610] handle sender defer: 4
219
+ 2024-12-10 20:54:35,581 INFO SenderThread:598006 [sender.py:transition_state():614] send defer: 5
220
+ 2024-12-10 20:54:35,581 DEBUG SenderThread:598006 [sender.py:send_request():406] send_request: poll_exit
221
+ 2024-12-10 20:54:35,582 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: defer
222
+ 2024-12-10 20:54:35,582 INFO HandlerThread:598006 [handler.py:handle_request_defer():184] handle defer: 5
223
+ 2024-12-10 20:54:35,582 DEBUG SenderThread:598006 [sender.py:send():379] send: summary
224
+ 2024-12-10 20:54:35,582 INFO SenderThread:598006 [sender.py:_save_file():1454] saving file wandb-summary.json with policy end
225
+ 2024-12-10 20:54:35,582 DEBUG SenderThread:598006 [sender.py:send_request():406] send_request: defer
226
+ 2024-12-10 20:54:35,582 INFO SenderThread:598006 [sender.py:send_request_defer():610] handle sender defer: 5
227
+ 2024-12-10 20:54:35,582 INFO SenderThread:598006 [sender.py:transition_state():614] send defer: 6
228
+ 2024-12-10 20:54:35,583 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: defer
229
+ 2024-12-10 20:54:35,583 INFO HandlerThread:598006 [handler.py:handle_request_defer():184] handle defer: 6
230
+ 2024-12-10 20:54:35,583 DEBUG SenderThread:598006 [sender.py:send_request():406] send_request: defer
231
+ 2024-12-10 20:54:35,583 INFO SenderThread:598006 [sender.py:send_request_defer():610] handle sender defer: 6
232
+ 2024-12-10 20:54:35,583 INFO SenderThread:598006 [sender.py:transition_state():614] send defer: 7
233
+ 2024-12-10 20:54:35,583 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: status_report
234
+ 2024-12-10 20:54:35,583 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: defer
235
+ 2024-12-10 20:54:35,583 INFO HandlerThread:598006 [handler.py:handle_request_defer():184] handle defer: 7
236
+ 2024-12-10 20:54:35,583 DEBUG SenderThread:598006 [sender.py:send_request():406] send_request: defer
237
+ 2024-12-10 20:54:35,583 INFO SenderThread:598006 [sender.py:send_request_defer():610] handle sender defer: 7
238
+ 2024-12-10 20:54:36,166 INFO Thread-12 :598006 [dir_watcher.py:_on_file_modified():288] file/dir modified: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205324/wandb/run-20241210_205330-52nddla7/files/wandb-summary.json
239
+ 2024-12-10 20:54:36,260 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: poll_exit
240
+ 2024-12-10 20:54:37,110 INFO SenderThread:598006 [sender.py:transition_state():614] send defer: 8
241
+ 2024-12-10 20:54:37,110 DEBUG SenderThread:598006 [sender.py:send_request():406] send_request: poll_exit
242
+ 2024-12-10 20:54:37,110 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: defer
243
+ 2024-12-10 20:54:37,110 INFO HandlerThread:598006 [handler.py:handle_request_defer():184] handle defer: 8
244
+ 2024-12-10 20:54:37,111 DEBUG SenderThread:598006 [sender.py:send_request():406] send_request: defer
245
+ 2024-12-10 20:54:37,111 INFO SenderThread:598006 [sender.py:send_request_defer():610] handle sender defer: 8
246
+ 2024-12-10 20:54:37,111 INFO SenderThread:598006 [job_builder.py:build():440] Attempting to build job artifact
247
+ 2024-12-10 20:54:37,113 INFO SenderThread:598006 [job_builder.py:_get_source_type():569] is repo sourced job
248
+ 2024-12-10 20:54:37,127 INFO SenderThread:598006 [job_builder.py:build():545] adding wandb-job metadata file
249
+ 2024-12-10 20:54:37,136 INFO SenderThread:598006 [sender.py:transition_state():614] send defer: 9
250
+ 2024-12-10 20:54:37,136 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: defer
251
+ 2024-12-10 20:54:37,136 INFO HandlerThread:598006 [handler.py:handle_request_defer():184] handle defer: 9
252
+ 2024-12-10 20:54:37,137 DEBUG SenderThread:598006 [sender.py:send():379] send: artifact
253
+ 2024-12-10 20:54:37,166 INFO Thread-12 :598006 [dir_watcher.py:_on_file_modified():288] file/dir modified: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205324/wandb/run-20241210_205330-52nddla7/files/output.log
254
+ 2024-12-10 20:54:37,260 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: poll_exit
255
+ 2024-12-10 20:54:37,780 INFO wandb-upload_1:598006 [upload_job.py:push():85] Skipped uploading /home/leonmkim/.local/share/wandb/artifacts/staging/tmp_p_p260o
256
+ 2024-12-10 20:54:37,783 INFO wandb-upload_0:598006 [upload_job.py:push():85] Skipped uploading /tmp/tmppldao70p/wandb-job.json
257
+ 2024-12-10 20:54:37,954 INFO wandb-upload_2:598006 [upload_job.py:push():88] Uploaded file /home/leonmkim/.local/share/wandb/artifacts/staging/tmp4q7vuxho
258
+ 2024-12-10 20:54:38,567 INFO SenderThread:598006 [sender.py:send_artifact():1537] sent artifact job-https___github.com_leonmkim_fish_leon.git_FISH_eval_robot.py - {'id': 'QXJ0aWZhY3Q6MTM2Njc2MTczOA==', 'state': 'PENDING', 'artifactSequence': {'id': 'QXJ0aWZhY3RDb2xsZWN0aW9uOjMxNzk5NjkyNQ==', 'latestArtifact': {'id': 'QXJ0aWZhY3Q6MTM2Njc1OTg1Ng==', 'versionIndex': 336}}}
259
+ 2024-12-10 20:54:38,567 DEBUG SenderThread:598006 [sender.py:send_request():406] send_request: defer
260
+ 2024-12-10 20:54:38,567 INFO SenderThread:598006 [sender.py:send_request_defer():610] handle sender defer: 9
261
+ 2024-12-10 20:54:38,567 INFO SenderThread:598006 [dir_watcher.py:finish():358] shutting down directory watcher
262
+ 2024-12-10 20:54:39,167 INFO SenderThread:598006 [dir_watcher.py:finish():388] scan: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205324/wandb/run-20241210_205330-52nddla7/files
263
+ 2024-12-10 20:54:39,168 INFO SenderThread:598006 [dir_watcher.py:finish():402] scan save: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205324/wandb/run-20241210_205330-52nddla7/files/config.yaml config.yaml
264
+ 2024-12-10 20:54:39,168 INFO SenderThread:598006 [dir_watcher.py:finish():402] scan save: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205324/wandb/run-20241210_205330-52nddla7/files/wandb-summary.json wandb-summary.json
265
+ 2024-12-10 20:54:39,168 INFO SenderThread:598006 [dir_watcher.py:finish():402] scan save: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205324/wandb/run-20241210_205330-52nddla7/files/events.out.tfevents.1733882011.leonmkim-ROG-Strix-G15CS-G15CS.597884.0 events.out.tfevents.1733882011.leonmkim-ROG-Strix-G15CS-G15CS.597884.0
266
+ 2024-12-10 20:54:39,168 INFO SenderThread:598006 [dir_watcher.py:finish():402] scan save: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205324/wandb/run-20241210_205330-52nddla7/files/requirements.txt requirements.txt
267
+ 2024-12-10 20:54:39,168 INFO SenderThread:598006 [dir_watcher.py:finish():402] scan save: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205324/wandb/run-20241210_205330-52nddla7/files/wandb-metadata.json wandb-metadata.json
268
+ 2024-12-10 20:54:39,168 INFO SenderThread:598006 [dir_watcher.py:finish():402] scan save: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205324/wandb/run-20241210_205330-52nddla7/files/output.log output.log
269
+ 2024-12-10 20:54:39,173 INFO SenderThread:598006 [dir_watcher.py:finish():402] scan save: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205324/wandb/run-20241210_205330-52nddla7/files/diff.patch diff.patch
270
+ 2024-12-10 20:54:39,173 INFO SenderThread:598006 [dir_watcher.py:finish():402] scan save: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205324/wandb/run-20241210_205330-52nddla7/files/code/FISH/eval_robot.py code/FISH/eval_robot.py
271
+ 2024-12-10 20:54:39,174 INFO SenderThread:598006 [dir_watcher.py:finish():402] scan save: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205324/wandb/run-20241210_205330-52nddla7/files/media/videos/eval/video_0_ff4edecb6a17ea039fbf.mp4 media/videos/eval/video_0_ff4edecb6a17ea039fbf.mp4
272
+ 2024-12-10 20:54:39,175 INFO SenderThread:598006 [sender.py:transition_state():614] send defer: 10
273
+ 2024-12-10 20:54:39,175 DEBUG SenderThread:598006 [sender.py:send_request():406] send_request: poll_exit
274
+ 2024-12-10 20:54:39,175 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: defer
275
+ 2024-12-10 20:54:39,175 INFO HandlerThread:598006 [handler.py:handle_request_defer():184] handle defer: 10
276
+ 2024-12-10 20:54:39,175 DEBUG SenderThread:598006 [sender.py:send_request():406] send_request: defer
277
+ 2024-12-10 20:54:39,175 INFO SenderThread:598006 [sender.py:send_request_defer():610] handle sender defer: 10
278
+ 2024-12-10 20:54:39,175 INFO SenderThread:598006 [file_pusher.py:finish():169] shutting down file pusher
279
+ 2024-12-10 20:54:39,261 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: poll_exit
280
+ 2024-12-10 20:54:39,261 DEBUG SenderThread:598006 [sender.py:send_request():406] send_request: poll_exit
281
+ 2024-12-10 20:54:39,363 INFO wandb-upload_0:598006 [upload_job.py:push():130] Uploaded file /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205324/wandb/run-20241210_205330-52nddla7/files/wandb-summary.json
282
+ 2024-12-10 20:54:39,397 INFO wandb-upload_2:598006 [upload_job.py:push():130] Uploaded file /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205324/wandb/run-20241210_205330-52nddla7/files/requirements.txt
283
+ 2024-12-10 20:54:39,416 INFO wandb-upload_1:598006 [upload_job.py:push():130] Uploaded file /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205324/wandb/run-20241210_205330-52nddla7/files/config.yaml
284
+ 2024-12-10 20:54:39,447 INFO wandb-upload_3:598006 [upload_job.py:push():130] Uploaded file /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205324/wandb/run-20241210_205330-52nddla7/files/output.log
285
+ 2024-12-10 20:54:39,647 INFO Thread-11 (_thread_body):598006 [sender.py:transition_state():614] send defer: 11
286
+ 2024-12-10 20:54:39,647 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: defer
287
+ 2024-12-10 20:54:39,647 INFO HandlerThread:598006 [handler.py:handle_request_defer():184] handle defer: 11
288
+ 2024-12-10 20:54:39,648 DEBUG SenderThread:598006 [sender.py:send_request():406] send_request: defer
289
+ 2024-12-10 20:54:39,648 INFO SenderThread:598006 [sender.py:send_request_defer():610] handle sender defer: 11
290
+ 2024-12-10 20:54:39,648 INFO SenderThread:598006 [file_pusher.py:join():175] waiting for file pusher
291
+ 2024-12-10 20:54:39,648 INFO SenderThread:598006 [sender.py:transition_state():614] send defer: 12
292
+ 2024-12-10 20:54:39,648 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: defer
293
+ 2024-12-10 20:54:39,648 INFO HandlerThread:598006 [handler.py:handle_request_defer():184] handle defer: 12
294
+ 2024-12-10 20:54:39,648 DEBUG SenderThread:598006 [sender.py:send_request():406] send_request: defer
295
+ 2024-12-10 20:54:39,649 INFO SenderThread:598006 [sender.py:send_request_defer():610] handle sender defer: 12
296
+ 2024-12-10 20:54:39,649 INFO SenderThread:598006 [file_stream.py:finish():601] file stream finish called
297
+ 2024-12-10 20:54:39,709 INFO SenderThread:598006 [file_stream.py:finish():605] file stream finish is done
298
+ 2024-12-10 20:54:39,709 INFO SenderThread:598006 [sender.py:transition_state():614] send defer: 13
299
+ 2024-12-10 20:54:39,709 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: defer
300
+ 2024-12-10 20:54:39,709 INFO HandlerThread:598006 [handler.py:handle_request_defer():184] handle defer: 13
301
+ 2024-12-10 20:54:39,709 DEBUG SenderThread:598006 [sender.py:send_request():406] send_request: defer
302
+ 2024-12-10 20:54:39,709 INFO SenderThread:598006 [sender.py:send_request_defer():610] handle sender defer: 13
303
+ 2024-12-10 20:54:39,709 INFO SenderThread:598006 [sender.py:transition_state():614] send defer: 14
304
+ 2024-12-10 20:54:39,709 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: defer
305
+ 2024-12-10 20:54:39,709 INFO HandlerThread:598006 [handler.py:handle_request_defer():184] handle defer: 14
306
+ 2024-12-10 20:54:39,710 DEBUG SenderThread:598006 [sender.py:send():379] send: final
307
+ 2024-12-10 20:54:39,710 DEBUG SenderThread:598006 [sender.py:send():379] send: footer
308
+ 2024-12-10 20:54:39,710 DEBUG SenderThread:598006 [sender.py:send_request():406] send_request: defer
309
+ 2024-12-10 20:54:39,710 INFO SenderThread:598006 [sender.py:send_request_defer():610] handle sender defer: 14
310
+ 2024-12-10 20:54:39,710 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: poll_exit
311
+ 2024-12-10 20:54:39,710 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: server_info
312
+ 2024-12-10 20:54:39,710 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: poll_exit
313
+ 2024-12-10 20:54:39,711 DEBUG SenderThread:598006 [sender.py:send_request():406] send_request: poll_exit
314
+ 2024-12-10 20:54:39,711 DEBUG SenderThread:598006 [sender.py:send_request():406] send_request: server_info
315
+ 2024-12-10 20:54:39,712 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: get_summary
316
+ 2024-12-10 20:54:39,713 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: sampled_history
317
+ 2024-12-10 20:54:39,713 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
318
+ 2024-12-10 20:54:39,793 DEBUG SenderThread:598006 [sender.py:send_request():406] send_request: poll_exit
319
+ 2024-12-10 20:54:39,794 INFO MainThread:598006 [wandb_run.py:_footer_history_summary_info():4016] rendering history
320
+ 2024-12-10 20:54:39,794 INFO MainThread:598006 [wandb_run.py:_footer_history_summary_info():4048] rendering summary
321
+ 2024-12-10 20:54:39,794 INFO MainThread:598006 [wandb_run.py:_footer_sync_info():3975] logging synced files
322
+ 2024-12-10 20:54:39,794 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: shutdown
323
+ 2024-12-10 20:54:39,794 INFO HandlerThread:598006 [handler.py:finish():882] shutting down handler
324
+ 2024-12-10 20:54:40,712 INFO WriterThread:598006 [datastore.py:close():296] close: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205324/wandb/run-20241210_205330-52nddla7/run-52nddla7.wandb
325
+ 2024-12-10 20:54:40,794 INFO SenderThread:598006 [sender.py:finish():1615] shutting down sender
326
+ 2024-12-10 20:54:40,794 INFO SenderThread:598006 [file_pusher.py:finish():169] shutting down file pusher
327
+ 2024-12-10 20:54:40,794 INFO SenderThread:598006 [file_pusher.py:join():175] waiting for file pusher
205324/wandb/debug.log ADDED
@@ -0,0 +1,31 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ 2024-12-10 20:53:30,845 INFO MainThread:597884 [wandb_setup.py:_flush():76] Current SDK version is 0.17.5
2
+ 2024-12-10 20:53:30,845 INFO MainThread:597884 [wandb_setup.py:_flush():76] Configure stats pid to 597884
3
+ 2024-12-10 20:53:30,845 INFO MainThread:597884 [wandb_setup.py:_flush():76] Loading settings from /home/leonmkim/.config/wandb/settings
4
+ 2024-12-10 20:53:30,845 INFO MainThread:597884 [wandb_setup.py:_flush():76] Loading settings from /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205324/wandb/settings
5
+ 2024-12-10 20:53:30,845 INFO MainThread:597884 [wandb_setup.py:_flush():76] Loading settings from environment variables: {}
6
+ 2024-12-10 20:53:30,845 INFO MainThread:597884 [wandb_setup.py:_flush():76] Applying setup settings: {'_disable_service': False}
7
+ 2024-12-10 20:53:30,845 INFO MainThread:597884 [wandb_setup.py:_flush():76] Inferring run settings from compute environment: {'program_relpath': 'FISH/eval_robot.py', 'program_abspath': '/home/leonmkim/fish_leon/FISH/eval_robot.py', 'program': '/home/leonmkim/fish_leon/FISH/eval_robot.py'}
8
+ 2024-12-10 20:53:30,845 INFO MainThread:597884 [wandb_setup.py:_flush():76] Applying login settings: {}
9
+ 2024-12-10 20:53:30,845 INFO MainThread:597884 [wandb_init.py:_log_setup():529] Logging user logs to /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205324/wandb/run-20241210_205330-52nddla7/logs/debug.log
10
+ 2024-12-10 20:53:30,845 INFO MainThread:597884 [wandb_init.py:_log_setup():530] Logging internal logs to /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205324/wandb/run-20241210_205330-52nddla7/logs/debug-internal.log
11
+ 2024-12-10 20:53:30,845 INFO MainThread:597884 [wandb_init.py:init():569] calling init triggers
12
+ 2024-12-10 20:53:30,845 INFO MainThread:597884 [wandb_init.py:init():576] wandb.init called with sweep_config: {}
13
+ config: {'root_dir': '/home/leonmkim/fish_leon', 'replay_buffer_size': 150000, 'replay_buffer_num_workers': 2, 'nstep': 3, 'batch_size': 128, 'seed': 0, 'dataset_shuffle_seed': 0, 'device': 'cuda', 'save_video': True, 'save_train_video': True, 'use_tb': True, 'use_wandb': True, 'wandb_run_id': '265626', 'wandb_notes': '265626_req_265827', 'eval': True, 'process_contact_features': True, 'obs_type': 'pixels', 'use_color': True, 'use_depth': True, 'use_masks': True, 'mask_list': ['EE_obj_mask'], 'mask_representation': 'channels', 'crop_hw': [144, 144], 'crop_down_offset': 48, 'color_crop_type': None, 'depth_crop_type': None, 'segmask_crop_type': None, 'add_crop_binary_mask': False, 'add_coord_conv_map': False, 'use_context_color': False, 'use_context_depth': False, 'use_context_segmask': False, 'context_color_crop_type': None, 'context_depth_crop_type': None, 'context_segmask_crop_type': None, 'context_add_crop_binary_mask': False, 'context_add_coord_conv_map': False, 'use_contact_map': False, 'use_sdf_maps': False, 'use_normals_maps': False, 'which_objects': 'both', 'max_contact_prob': 0.1, 'max_depth': 2.0, 'grasped_dtc_max_value': 0.2, 'env_dtc_max_value': 0.4, 'grasped_normals_mask_max_dtc_value': 0.2, 'env_normals_mask_max_dtc_value': 0.4, 'clamp_dtc': True, 'dtc_adaptive_normalization': False, 'mask_normals_within_sdf': True, 'adaptive_normals_mask': True, 'learnable_contact_preprocess_params': True, 'contact_model_name': 'local_multitask_outhd64all_home_crop_h144w144d48_mask_ctxtmask_seed_220979_epoch_9', 'contact_estimation_model_ckpt_path': '~/fish_leon/contact_estimation/artifacts/175604_2/checkpoints/epoch=09-val_loss=0.00.ckpt', 'encoder_type': 'small', 'debug_timestamps': False, 'open_loop': False, 'action_trajectories': True, 'stop_after_action': False, 'interpolation_frequency': 25, 'policy_frequency': 5, 'wait_for_new_camera_frames': True, 'baseline': False, 'train_demo_idxs_list_or_num': -1, 'log_train_every_steps': 25, 'name_of_expert_demo': '64_240x320_hbm_twodim_fps_fix_contact_fix_recovery_fix_annotated_start_idx_5hz_zstd7_EE_pxl_coords_expert_demos_imp_act', 'expert_dataset_dirpath': '/home/leonmkim/fish_leon/FISH/expert_demos/frankagym/FrankaInsertion-v1/64_240x320_hbm_twodim_fps_fix_contact_fix_recovery_fix_annotated_start_idx_5hz_zstd7_EE_pxl_coords_expert_demos_imp_act', 'store_dataset_in_memory': False, 'expert_dataset': '/home/leonmkim/fish_leon/FISH/expert_demos/frankagym/FrankaInsertion-v1/64_240x320_hbm_twodim_fps_fix_contact_fix_recovery_fix_annotated_start_idx_5hz_zstd7_EE_pxl_coords_expert_demos_imp_act/demos.zarr', 'action_key': 'action_trajectory_25hz', 'semantic_demo_grouping_name': 'semantic_demo_grouping.yaml', 'semantic_demo_grouping': '/home/leonmkim/fish_leon/FISH/expert_demos/frankagym/FrankaInsertion-v1/64_240x320_hbm_twodim_fps_fix_contact_fix_recovery_fix_annotated_start_idx_5hz_zstd7_EE_pxl_coords_expert_demos_imp_act/semantic_demo_grouping.yaml', 'include_groups_list': ['hbm_twodim_left_to_right'], 'expert_dataset_config': '/home/leonmkim/fish_leon/FISH/expert_demos/frankagym/FrankaInsertion-v1/64_240x320_hbm_twodim_fps_fix_contact_fix_recovery_fix_annotated_start_idx_5hz_zstd7_EE_pxl_coords_expert_demos_imp_act/demo_config.yaml', 'name_of_valid_demo': '64_240x320_hbm_twodim_fps_fix_contact_fix_recovery_fix_annotated_start_idx_5hz_zstd7_EE_pxl_coords_expert_demos_imp_act', 'valid_dataset_dir': '/home/leonmkim/fish_leon/FISH/expert_demos/frankagym/FrankaInsertion-v1/64_240x320_hbm_twodim_fps_fix_contact_fix_recovery_fix_annotated_start_idx_5hz_zstd7_EE_pxl_coords_expert_demos_imp_act/demos.zarr', 'valid_demo_idxs_list_or_num': None, 'val_num_groups': 0, 'load_bc': True, 'checkpoint_epoch_list': [99, 199, 299, 399, 499, 599, 699, 799, 899, 999, 1249, 1499, 1749, 1999, 2999, 3999, 4999, 5999, 6999, 7999, 8999, 9999], 'snapshot_root_dir': '/mnt/grasp_high_usage/leonmkim/contact_estimation/FISH', 'save_snapshot': True, 'save_last_snapshot': True, 'save_snapshot_when_done': True, 'top_k_checkpoints': 5, 'save_snapshot_link_to_weights_dir': 'deprecated', 'bc_regularize': False, 'bc_weight_type': 'qfilter', 'experiment_dir': './exp_local/frankagym_pixels/FrankaInsertion-v1/265626', 'agent': {'_target_': 'agent.diffusion_policy.DiffusionPolicyAgent', 'name': 'diffusion_policy', 'load_checkpoint': True, 'obs_shape': [5, 240, 320], 'action_shape': [36, 7], 'device': 'cuda', 'n_obs_steps': 1, 'suite_name': 'frankagym', 'obs_type': 'pixels', 'enable_arm': True, 'enable_camera': True, 'use_tb': True, 'desired_image_shape': [13, 180, 240], 'orig_cam_shape': [3, 240, 320], 'config': {'_target_': 'agent.diffusion_policy.DiffusionPolicyAgentConfig', 'compile': False, 'device': 'cuda', 'cam_resize_shape': [13, 180, 240], 'orig_cam_shape': [3, 240, 320], 'policy_cfg': {'_target_': 'lerobot.common.policies.diffusion.configuration_diffusion.DiffusionConfig', 'n_obs_steps': 1, 'horizon': 36, 'n_action_steps': 36, 'output_shapes': {'action': [7]}, 'input_normalization_modes': {'observation.image': 'mean_std', 'observation.state': 'min_max', 'observation.action_history': 'min_max'}, 'output_normalization_modes': {'action': 'min_max'}, 'vision_backbone': 'resnet18', 'pretrained_backbone_weights': None, 'transforms': [{'_target_': 'torchaug.transforms.RandomAffine', 'degrees': [-5, 5], 'translate': [0.05, 0.05], 'batch_transform': True, 'num_chunks': -1, 'batch_inplace': True}, {'_target_': 'torchaug.transforms.RandomColorJitter', 'brightness': 0.3, 'contrast': 0.4, 'saturation': 0.5, 'hue': 0.08, 'batch_transform': True, 'num_chunks': -1, 'batch_inplace': True}], 'use_group_norm': True, 'spatial_softmax_num_keypoints': 32, 'action_history_encoder_config': {'_target_': 'lerobot.common.policies.diffusion.configuration_diffusion.Unet1dEncoderConfig', 'in_channels': 7, 'out_channels': 32, 'history_length': 6, 'kernel_size': 5, 'downsample_kernel_size': 3, 'downsample_stride': 2, 'downsample_padding': 1}, 'down_dims': [256, 512, 1024], 'kernel_size': 5, 'n_groups': 8, 'diffusion_step_embed_dim': 128, 'use_film_scale_modulation': True, 'noise_scheduler_type': 'DDIM', 'beta_schedule': 'squaredcos_cap_v2', 'beta_start': 0.0001, 'beta_end': 0.02, 'prediction_type': 'epsilon', 'clip_sample': True, 'clip_sample_range': 1.0, 'num_train_timesteps': 50, 'num_inference_steps': 10, 'do_mask_loss_for_padding': False, 'input_shapes': {'observation.image': [13, 180, 240], 'context_observation.image': [13, 180, 240], 'observation.state': [8], 'observation.action_history': [7]}}, 'train_cfg': {'_target_': 'utils.TrainConfig', 'lr': 0.0001, 'lr_scheduler': 'cosine', 'lr_warmup_steps': 500, 'adam_betas': [0.95, 0.999], 'adam_eps': 1e-08, 'adam_weight_decay': 1e-06, 'grad_clip_norm': 10, 'offline_steps': 1000000, 'use_amp': True}, 'observation_cfg': {'_target_': 'agent.encoder.VisualFeatureSet', 'use_depth': True, 'use_color': True, 'mask_input_dict': {'_target_': 'agent.encoder.MaskInputDict', 'enable': True, 'representation': 'channels', 'mask_list': ['EE_obj_mask']}, 'crop_input_config': {'_target_': 'agent.encoder.CropInputConfig', 'color_crop_type': None, 'depth_crop_type': None, 'segmask_crop_type': None, 'crop_hw': [144, 144], 'crop_down_offset': 48, 'add_crop_binary_mask': False, 'add_coord_conv_map': False}, 'context_input_config': {'_target_': 'agent.encoder.ContextInputConfig', 'use_color': False, 'use_depth': False, 'mask_input_dict': {'_target_': 'agent.encoder.MaskInputDict', 'enable': False, 'representation': 'channels', 'mask_list': ['EE_obj_mask']}, 'crop_input_config': {'_target_': 'agent.encoder.CropInputConfig', 'color_crop_type': None, 'depth_crop_type': None, 'segmask_crop_type': None, 'crop_hw': [144, 144], 'crop_down_offset': 48, 'add_crop_binary_mask': False, 'add_coord_conv_map': False}}, 'mask_soft_approx_scheduler_config': {'_target_': 'agent.encoder.MaskSoftApproxSchedulerConfig', 'num_steps': 40000, 'initial_value': 10.0, 'final_value': 1000.0, 'interpolation_scheme': 'cosine'}, 'use_contact_map': False, 'use_sdf_maps': False, 'use_normals_maps': False, 'which_objects': 'both', 'grasped_dtc_max_value': 0.2, 'env_dtc_max_value': 0.4, 'grasped_normals_mask_max_dtc_value': 0.2, 'env_normals_mask_max_dtc_value': 0.4, 'clamp_dtc': True, 'max_contact_prob': 0.1, 'mask_normals_within_sdf': True, 'dtc_adaptive_normalization': False, 'adaptive_normals_mask': True, 'max_depth': 2.0, 'image_shape': [13, 180, 240], 'learnable_contact_preprocess_params': True, 'learning_rate': 0.0001, 'weight_decay': 0.0, 'contact_model_name': 'local_multitask_outhd64all_home_crop_h144w144d48_mask_ctxtmask_seed_220979_epoch_9', 'zero_centered': False}, 'policy_frequency': 5, 'interpolation_frequency': 25}}, 'suite': {'suite': 'frankagym', 'name': 'frankagym', 'frame_stack': 1, 'action_repeat': 1, 'discount': 0.99, 'hidden_dim': 1024, 'num_train_frames': 2010, 'num_seed_frames': 260, 'num_train_epochs': 5000, 'validate_every_epochs': 100, 'validate_diffusion_on_action_loss_every_epochs': 500, 'train_eval_diffusion_on_action_loss_every_epochs': 500, 'check_topk_every_epochs': 10, 'save_snapshot_every_epochs': 5000, 'eval_every_frames': 2000, 'num_eval_episodes': 5, 'save_snapshot': True, 'wait_for_user_to_start_episode': True, 'task_make_fn': {'_target_': 'suite.frankagym.make', 'name': 'FrankaInsertion-v1', 'height': 240, 'width': 320, 'frame_stack': 1, 'action_repeat': 1, 'seed': 0, 'enable_arm': True, 'enable_gripper': True, 'start_with_gripper_open': True, 'enable_camera': True, 'path_to_depth_extrinsics': '~/fish_leon/FISH/cfgs/camera_poses/camera_poses_L515/20240904-122305/color_tf_world.npy', 'contact_estimation_model_ckpt_path': '~/fish_leon/contact_estimation/artifacts/175604_2/checkpoints/epoch=09-val_loss=0.00.ckpt', 'x_limit': [0.2, 0.7], 'y_limit': [-0.4, 0.4], 'z_limit': [-0.05, 0.55], 'device': 'cuda', 'interpolation_frequency': 25, 'policy_frequency': 5, 'debug_timestamps': False, 'stop_after_action': False, 'open_loop': False, 'wait_for_new_camera_frames': True, 'action_key': 'action_trajectory_25hz', 'action_trajectory_horizon': 36, 'action_trajectories': True, 'path_to_zarr_dataset': '/home/leonmkim/fish_leon/FISH/expert_demos/frankagym/FrankaInsertion-v1/64_240x320_hbm_twodim_fps_fix_contact_fix_recovery_fix_annotated_start_idx_5hz_zstd7_EE_pxl_coords_expert_demos_imp_act/demos.zarr', 'observation_cfg': {'_target_': 'agent.encoder.VisualFeatureSet', 'use_depth': True, 'use_color': True, 'mask_input_dict': {'_target_': 'agent.encoder.MaskInputDict', 'enable': True, 'representation': 'channels', 'mask_list': ['EE_obj_mask']}, 'crop_input_config': {'_target_': 'agent.encoder.CropInputConfig', 'color_crop_type': None, 'depth_crop_type': None, 'segmask_crop_type': None, 'crop_hw': [144, 144], 'crop_down_offset': 48, 'add_crop_binary_mask': False, 'add_coord_conv_map': False}, 'context_input_config': {'_target_': 'agent.encoder.ContextInputConfig', 'use_color': False, 'use_depth': False, 'mask_input_dict': {'_target_': 'agent.encoder.MaskInputDict', 'enable': False, 'representation': 'channels', 'mask_list': ['EE_obj_mask']}, 'crop_input_config': {'_target_': 'agent.encoder.CropInputConfig', 'color_crop_type': None, 'depth_crop_type': None, 'segmask_crop_type': None, 'crop_hw': [144, 144], 'crop_down_offset': 48, 'add_crop_binary_mask': False, 'add_coord_conv_map': False}}, 'mask_soft_approx_scheduler_config': {'_target_': 'agent.encoder.MaskSoftApproxSchedulerConfig', 'num_steps': 40000, 'initial_value': 10.0, 'final_value': 1000.0, 'interpolation_scheme': 'cosine'}, 'use_contact_map': False, 'use_sdf_maps': False, 'use_normals_maps': False, 'which_objects': 'both', 'grasped_dtc_max_value': 0.2, 'env_dtc_max_value': 0.4, 'grasped_normals_mask_max_dtc_value': 0.2, 'env_normals_mask_max_dtc_value': 0.4, 'clamp_dtc': True, 'max_contact_prob': 0.1, 'mask_normals_within_sdf': True, 'dtc_adaptive_normalization': False, 'adaptive_normals_mask': True, 'max_depth': 2.0, 'image_shape': [13, 180, 240], 'learnable_contact_preprocess_params': True, 'learning_rate': 0.0001, 'weight_decay': 0.0, 'contact_model_name': 'local_multitask_outhd64all_home_crop_h144w144d48_mask_ctxtmask_seed_220979_epoch_9', 'zero_centered': False}, 'agent_policy_cfg': {'_target_': 'agent.diffusion_policy.DiffusionPolicyAgentConfig', 'compile': False, 'device': 'cuda', 'cam_resize_shape': [13, 180, 240], 'orig_cam_shape': [3, 240, 320], 'policy_cfg': {'_target_': 'lerobot.common.policies.diffusion.configuration_diffusion.DiffusionConfig', 'n_obs_steps': 1, 'horizon': 36, 'n_action_steps': 36, 'output_shapes': {'action': [7]}, 'input_normalization_modes': {'observation.image': 'mean_std', 'observation.state': 'min_max', 'observation.action_history': 'min_max'}, 'output_normalization_modes': {'action': 'min_max'}, 'vision_backbone': 'resnet18', 'pretrained_backbone_weights': None, 'transforms': [{'_target_': 'torchaug.transforms.RandomAffine', 'degrees': [-5, 5], 'translate': [0.05, 0.05], 'batch_transform': True, 'num_chunks': -1, 'batch_inplace': True}, {'_target_': 'torchaug.transforms.RandomColorJitter', 'brightness': 0.3, 'contrast': 0.4, 'saturation': 0.5, 'hue': 0.08, 'batch_transform': True, 'num_chunks': -1, 'batch_inplace': True}], 'use_group_norm': True, 'spatial_softmax_num_keypoints': 32, 'action_history_encoder_config': {'_target_': 'lerobot.common.policies.diffusion.configuration_diffusion.Unet1dEncoderConfig', 'in_channels': 7, 'out_channels': 32, 'history_length': 6, 'kernel_size': 5, 'downsample_kernel_size': 3, 'downsample_stride': 2, 'downsample_padding': 1}, 'down_dims': [256, 512, 1024], 'kernel_size': 5, 'n_groups': 8, 'diffusion_step_embed_dim': 128, 'use_film_scale_modulation': True, 'noise_scheduler_type': 'DDIM', 'beta_schedule': 'squaredcos_cap_v2', 'beta_start': 0.0001, 'beta_end': 0.02, 'prediction_type': 'epsilon', 'clip_sample': True, 'clip_sample_range': 1.0, 'num_train_timesteps': 50, 'num_inference_steps': 10, 'do_mask_loss_for_padding': False, 'input_shapes': {'observation.image': [13, 180, 240], 'context_observation.image': [13, 180, 240], 'observation.state': [8], 'observation.action_history': [7]}}, 'train_cfg': {'_target_': 'utils.TrainConfig', 'lr': 0.0001, 'lr_scheduler': 'cosine', 'lr_warmup_steps': 500, 'adam_betas': [0.95, 0.999], 'adam_eps': 1e-08, 'adam_weight_decay': 1e-06, 'grad_clip_norm': 10, 'offline_steps': 1000000, 'use_amp': True}, 'observation_cfg': {'_target_': 'agent.encoder.VisualFeatureSet', 'use_depth': True, 'use_color': True, 'mask_input_dict': {'_target_': 'agent.encoder.MaskInputDict', 'enable': True, 'representation': 'channels', 'mask_list': ['EE_obj_mask']}, 'crop_input_config': {'_target_': 'agent.encoder.CropInputConfig', 'color_crop_type': None, 'depth_crop_type': None, 'segmask_crop_type': None, 'crop_hw': [144, 144], 'crop_down_offset': 48, 'add_crop_binary_mask': False, 'add_coord_conv_map': False}, 'context_input_config': {'_target_': 'agent.encoder.ContextInputConfig', 'use_color': False, 'use_depth': False, 'mask_input_dict': {'_target_': 'agent.encoder.MaskInputDict', 'enable': False, 'representation': 'channels', 'mask_list': ['EE_obj_mask']}, 'crop_input_config': {'_target_': 'agent.encoder.CropInputConfig', 'color_crop_type': None, 'depth_crop_type': None, 'segmask_crop_type': None, 'crop_hw': [144, 144], 'crop_down_offset': 48, 'add_crop_binary_mask': False, 'add_coord_conv_map': False}}, 'mask_soft_approx_scheduler_config': {'_target_': 'agent.encoder.MaskSoftApproxSchedulerConfig', 'num_steps': 40000, 'initial_value': 10.0, 'final_value': 1000.0, 'interpolation_scheme': 'cosine'}, 'use_contact_map': False, 'use_sdf_maps': False, 'use_normals_maps': False, 'which_objects': 'both', 'grasped_dtc_max_value': 0.2, 'env_dtc_max_value': 0.4, 'grasped_normals_mask_max_dtc_value': 0.2, 'env_normals_mask_max_dtc_value': 0.4, 'clamp_dtc': True, 'max_contact_prob': 0.1, 'mask_normals_within_sdf': True, 'dtc_adaptive_normalization': False, 'adaptive_normals_mask': True, 'max_depth': 2.0, 'image_shape': [13, 180, 240], 'learnable_contact_preprocess_params': True, 'learning_rate': 0.0001, 'weight_decay': 0.0, 'contact_model_name': 'local_multitask_outhd64all_home_crop_h144w144d48_mask_ctxtmask_seed_220979_epoch_9', 'zero_centered': False}, 'policy_frequency': 5, 'interpolation_frequency': 25}, 'true_action_history': False}}, 'num_train_frames_bc': 50000, 'num_train_frames_drq': 1100000, 'stddev_schedule_drq': 'linear(1.0,0.1,100000)', 'task_name': 'FrankaInsertion-v1', 'num_train_frames_vinn': 25000, 'num_train_frames_diffusion': 1000000, 'num_train_epochs_bc': 5000, 'num_train_epochs_diffusion': 15000, 'validate_every_epochs_bc': 5, 'validate_every_epochs_diffusion': 250, 'validate_diffusion_on_action_loss_every_epochs': 250, 'train_eval_diffusion_on_action_loss_every_epochs': 250, 'check_topk_every_epochs': 5, 'check_topk_every_epochs_diffusion': 250, 'save_snapshot_every_epochs_diffusion': 1500, 'x_limit': [0.2, 0.7], 'y_limit': [-0.4, 0.4], 'z_limit': [-0.05, 0.55], 'home_displacement': [0.55, 0.0, 0.55, 180.0, 0.0, 0.0], 'enable_gripper': True, 'start_with_gripper_open': True, 'offset_mask': [1, 1, 1, 1, 1, 1], 'path_to_depth_extrinsics': '~/fish_leon/FISH/cfgs/camera_poses/camera_poses_L515/20240904-122305/color_tf_world.npy', 'feature_type': '180x240_1_RGB_D_2.0_msk_channels_EE_obj_mask_acthst_hst6_out32_dwnkrnl3_dwnstrd2_dwnpd1', 'save_buffer': True, 'num_eval': 5, 'random_start': False, 'eval_starts': '/home/leonmkim/fish_leon/FISH/eval_starts/frankagym_pixels/FrankaInsertion-v1', 'num_valid_demos': None, 'load_checkpoint': True, 'true_action_history': False, 'checkpoint_epoch': 12000, 'load_residual_weight': False, 'checkpoint_root_dir': '/home/leonmkim/fish_leon/FISH', 'checkpoint_weight_dir': '/home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626', 'residual_weight': '/home/leonmkim/fish_leon/FISH/weights/frankagym_pixels/FrankaInsertion-v1/weight.pt'}
14
+ 2024-12-10 20:53:30,845 INFO MainThread:597884 [wandb_init.py:init():619] starting backend
15
+ 2024-12-10 20:53:30,845 INFO MainThread:597884 [wandb_init.py:init():623] setting up manager
16
+ 2024-12-10 20:53:30,849 INFO MainThread:597884 [backend.py:_multiprocessing_setup():105] multiprocessing start_methods=fork,spawn,forkserver, using: spawn
17
+ 2024-12-10 20:53:30,850 INFO MainThread:597884 [wandb_init.py:init():631] backend started and connected
18
+ 2024-12-10 20:53:30,862 INFO MainThread:597884 [wandb_init.py:init():720] updated telemetry
19
+ 2024-12-10 20:53:30,871 INFO MainThread:597884 [wandb_init.py:init():753] communicating run to backend with 90.0 second timeout
20
+ 2024-12-10 20:53:31,160 INFO MainThread:597884 [wandb_run.py:_on_init():2435] communicating current version
21
+ 2024-12-10 20:53:31,200 INFO MainThread:597884 [wandb_run.py:_on_init():2444] got version response upgrade_message: "wandb version 0.19.0 is available! To upgrade, please run:\n $ pip install wandb --upgrade"
22
+
23
+ 2024-12-10 20:53:31,201 INFO MainThread:597884 [wandb_init.py:init():804] starting run threads in backend
24
+ 2024-12-10 20:53:31,571 INFO MainThread:597884 [wandb_run.py:_console_start():2413] atexit reg
25
+ 2024-12-10 20:53:31,571 INFO MainThread:597884 [wandb_run.py:_redirect():2255] redirect: wrap_raw
26
+ 2024-12-10 20:53:31,571 INFO MainThread:597884 [wandb_run.py:_redirect():2320] Wrapping output streams.
27
+ 2024-12-10 20:53:31,571 INFO MainThread:597884 [wandb_run.py:_redirect():2345] Redirects installed.
28
+ 2024-12-10 20:53:31,573 INFO MainThread:597884 [wandb_init.py:init():847] run started, returning control to user process
29
+ 2024-12-10 20:53:31,573 INFO MainThread:597884 [wandb_run.py:_tensorboard_callback():1544] tensorboard callback: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205324/tb, True
30
+ 2024-12-10 20:53:35,546 INFO MainThread:597884 [wandb_run.py:_config_callback():1382] config_cb None None {'grasped_obj_name': 'hbm', 'left_book_slot': 'twodim'}
31
+ 2024-12-10 20:54:40,795 WARNING MsgRouterThr:597884 [router.py:message_loop():77] message_loop has been closed
205324/wandb/run-20241210_205330-52nddla7/files/code/FISH/eval_robot.py ADDED
@@ -0,0 +1,512 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ #%%
2
+ import warnings
3
+ import os
4
+
5
+ os.environ['MKL_SERVICE_FORCE_INTEL'] = '1'
6
+ os.environ['MUJOCO_GL'] = 'egl'
7
+ from pathlib import Path
8
+ #%%
9
+ import hydra
10
+ import numpy as np
11
+ import torch
12
+
13
+ import utils
14
+ from utils import get_feature_dirname_from_configs
15
+
16
+ from video import VideoRecorder
17
+ import pickle
18
+ import time
19
+ import threading
20
+ import shutil
21
+ from logger import Logger
22
+
23
+ import wandb
24
+ from omegaconf import OmegaConf, open_dict
25
+
26
+ from replay_buffer_robot import RosbagEvalReplayBufferStorage
27
+ from lerobot.common.utils.utils import _relative_path_between
28
+
29
+ torch.backends.cudnn.benchmark = True
30
+ warnings.filterwarnings('ignore', category=DeprecationWarning)
31
+
32
+ # import specs for replay buffer
33
+ from dm_env import specs
34
+
35
+ import sys, signal
36
+ import yaml
37
+
38
+ # get path of current file
39
+ current_path = os.path.dirname(os.path.realpath(__file__))
40
+ sys.path.append(os.path.join(current_path, os.pardir))
41
+ # from contact_estimation.src.utils.viz_utils import normalized_surface_normal_to_rgb, depth_map_to_im, grasped_env_dtc_map_to_im, contact_prob_map_to_im, desaturate_color_image, masked_overlay_im_list
42
+
43
+ def make_agent(obs_spec, action_spec, cfg):
44
+ cfg.obs_shape = obs_spec['pixels'].shape
45
+ dataset_statistics = None # this will be loaded from the checkpoint
46
+ try:
47
+ cfg.action_shape = action_spec.shape
48
+ except:
49
+ pass
50
+ return hydra.utils.instantiate(cfg, dataset_statistics)
51
+
52
+ class Workspace:
53
+ def __init__(self, cfg):
54
+ self.work_dir = Path.cwd()
55
+ print(f'workspace: {self.work_dir}')
56
+
57
+ signal.signal(signal.SIGINT, self.signal_handler)
58
+
59
+ self.cfg = cfg
60
+ self.loading_uncompiled_checkpoint_with_compile = False
61
+ self.loading_compiled_checkpoint_with_no_compile = False
62
+
63
+ snapshot_path = Path(self.cfg.checkpoint_weight_dir) / f'snapshot_{self.cfg.checkpoint_epoch}.pt'
64
+ self.load_checkpoint_conf(snapshot_path=snapshot_path)
65
+
66
+ # load config for action trajectories
67
+ utils.set_seed_everywhere(self.cfg.seed)
68
+ self.device = torch.device(self.cfg.device)
69
+ self.setup()
70
+
71
+ # self.agent = make_agent(self.eval_env.observation_spec(),
72
+ # self.eval_env.action_spec(), self.cfg.agent)
73
+ self.timer = utils.Timer()
74
+ # self._global_step = 0
75
+ self._global_episode = 0
76
+ self._global_epoch = 0
77
+ self.num_episode_successes = 0
78
+
79
+ # Need to convert hydra config to primitive container for wandb https://docs.wandb.ai/guides/integrations/hydra
80
+ with open_dict(self.cfg):
81
+ self.cfg.feature_type = get_feature_dirname_from_configs(
82
+ hydra.utils.instantiate(self.cfg.agent.config.observation_cfg),
83
+ self.cfg.agent.config.policy_cfg.input_shapes,
84
+ hydra.utils.instantiate(self.cfg.agent.config.policy_cfg.action_history_encoder_config) if 'observation.action_history' in self.cfg.agent.config.policy_cfg.input_shapes else None,
85
+ )
86
+
87
+ wandb_config = OmegaConf.to_container(
88
+ self.cfg, resolve=True, throw_on_missing=True
89
+ )
90
+ # must be called before any tf summary writer is created
91
+ if self.cfg.use_wandb:
92
+ wandb.init(project='extrinsic_contact_downstream', entity='serialexperimentsleon', job_type='eval', sync_tensorboard=self.cfg.use_tb, config=wandb_config)
93
+
94
+ self.logger = Logger(self.work_dir, use_tb=self.cfg.use_tb, use_wandb=self.cfg.use_wandb)
95
+
96
+ # if not self.loading_uncompiled_checkpoint_with_compile and self.cfg.agent.config.compile:
97
+ # self.agent.compile_modules()
98
+
99
+ # self.load_checkpoint(snapshot_path=snapshot_path)
100
+
101
+ # if self.loading_uncompiled_checkpoint_with_compile: # need to call compile after loading the checkpoint
102
+ # self.agent.compile_modules()
103
+
104
+ print(f"loaded agent with feature_type: {self.cfg.feature_type}")
105
+
106
+ def check_for_key_press(self):
107
+ while self.continue_keypress_thread:
108
+ inp = input("Press 'r' to restart current episode, 'n' to stop current episode and skip to next, 'q' to break entire eval\n")
109
+ if inp == 'n':
110
+ self.preempt_episode = True
111
+ print("preempting episode")
112
+ elif inp in ['', '0', '1']: # enter key
113
+ if inp in ['0', '1']:
114
+ self.num_episode_successes += int(inp)
115
+ self.proceed_after_env_reset_event.set()
116
+ print("proceeding to start episode!")
117
+ elif inp == 'q':
118
+ self.proceed_after_env_reset_event.set()
119
+ self.preempt_episode = True
120
+ self.exit_eval = True
121
+ self.continue_keypress_thread = False # will stop the keypress thread
122
+ print("quitting eval")
123
+ break
124
+ elif inp == 'r':
125
+ print('restarting episode')
126
+ self.preempt_episode = True
127
+ self.restart_episode = True
128
+ else:
129
+ print("Invalid key press, try again")
130
+
131
+ # self.keypress_input_thread.join() # wait for the keypress thread to finish
132
+
133
+ def signal_handler(self, signal, frame):
134
+ print("\nprogram exiting gracefully")
135
+ self.proceed_after_env_reset_event.set()
136
+ self.preempt_episode = True
137
+ self.exit_eval = True
138
+ self.continue_keypress_thread = False # will stop the keypress thread
139
+ self.keypress_input_thread.join() # wait for the keypress thread to finish
140
+ video_filepath = self.video_recorder.save()
141
+ # get the video file and convert to video tensor to log
142
+ self.logger.log_video('eval/video', video_filepath, self.global_step)
143
+ sys.exit(0)
144
+
145
+ def setup(self):
146
+ # create envs
147
+ self.eval_env = hydra.utils.call(self.cfg.suite.task_make_fn)
148
+ # expert_demo_config_path = os.path.join(os.path.dirname(self.cfg.expert_dataset), 'demo_config.yaml')
149
+ # self.expert_demo_config = yaml.load(open(expert_demo_config_path, 'r'), Loader=yaml.FullLoader)
150
+ # self.eval_env._env.action_trans_norm = expert_demo_config['max_translation_action_norm']
151
+ # self.eval_env._env.action_rot_norm = expert_demo_config['max_rotation_action_norm']
152
+ # self.eval_env._env.action_period = expert_demo_config['sample_period']
153
+ # print(f"setting max_translation_action_norm to {expert_demo_config['max_translation_action_norm']} and sample_period to {expert_demo_config['sample_period']}")
154
+ # print(f"setting max_rotation_action_norm to {expert_demo_config['max_rotation_action_norm']}")
155
+
156
+ # self.eval_env.set_demo_params(self.cfg.expert_dataset)
157
+
158
+ # Turn off random start
159
+ self.eval_env.random_start = False
160
+
161
+ # create replay buffer
162
+ # data_specs = [
163
+ # {
164
+ # 'observation': self.eval_env.observation_spec(),
165
+ # },
166
+ # # self.eval_env.observation_spec()['features'],
167
+ # self.eval_env.action_spec(),
168
+ # specs.Array(self.eval_env.action_spec().shape, self.eval_env.action_spec().dtype, 'vinn_action'),
169
+ # specs.Array((1, ), np.float32, 'reward'),
170
+ # specs.Array((1, ), np.float32, 'discount'),
171
+ # ]
172
+
173
+ # self.eval_replay_storage = ZarrEvalReplayBufferStorage(data_specs, self.work_dir / 'eval_buffer', debug_timestamps=self.cfg.debug_timestamps, save_buffer=self.cfg.save_buffer, debug_info_data_specs=self.eval_env.debug_info_data_specs, camera_info_dict=self.eval_env.get_camera_info_dict())
174
+ self.eval_replay_storage = RosbagEvalReplayBufferStorage(self.work_dir)
175
+
176
+ self.video_recorder = VideoRecorder(
177
+ self.work_dir if self.cfg.save_video else None,
178
+ ros_enabled=True,
179
+ fps=self.cfg.agent.config.policy_frequency,
180
+ )
181
+
182
+ print('workspace setup complete')
183
+
184
+ @property
185
+ def global_step(self):
186
+ # return self._global_step
187
+ return self.eval_env.get_global_step()
188
+
189
+ @property
190
+ def global_episode(self):
191
+ return self._global_episode
192
+
193
+ @property
194
+ def global_frame(self):
195
+ return self.global_step * self.cfg.action_repeat
196
+
197
+ @property
198
+ def global_epoch(self):
199
+ return self._global_epoch
200
+
201
+ def reset(self, eval_idx):
202
+ if not self.eval_env.enable_arm:
203
+ return np.array([0,0,0], dtype=np.float32)
204
+ self.eval_env.arm_refresh(reset=False)
205
+ # Set start position
206
+ try:
207
+ self.eval_env.set_position(self.start_pos[eval_idx])
208
+ except:
209
+ self.eval_env.arm.set_position(self.start_pos[eval_idx])
210
+ if self.eval_env.arm.keep_gripper_closed:
211
+ self.eval_env.arm.close_gripper_fully()
212
+ else:
213
+ self.eval_env.arm.open_gripper_fully()
214
+ time.sleep(0.1)
215
+ time_step = self.eval_env.step(np.zeros(self.eval_env.action_spec().shape[0], dtype=np.float32),
216
+ np.zeros(self.eval_env.action_spec().shape[0], dtype=np.float32))
217
+ return time_step
218
+
219
+ def eval(self):
220
+ # before evals start, prompt user for name of grasped object and the left book of the slot location
221
+ grasped_obj_name = input("Enter the name of the grasped object: ")
222
+ left_book_slot = input("Enter the left book slot location: ")
223
+ # update wandb config
224
+ if self.cfg.use_wandb:
225
+ wandb.config.update({'grasped_obj_name': grasped_obj_name, 'left_book_slot': left_book_slot})
226
+
227
+ self.preempt_episode = False
228
+ self.exit_eval = False
229
+ self.restart_episode = False
230
+
231
+ self.continue_keypress_thread = True
232
+ self.proceed_after_env_reset_event = threading.Event()
233
+ self.keypress_input_thread = threading.Thread(target=self.check_for_key_press)
234
+ self.keypress_input_thread.start()
235
+
236
+ # # Set model to eval mode
237
+ # self.agent.train(False)
238
+
239
+ eval_until_episode = utils.Until(self.cfg.num_eval)
240
+
241
+ self.use_action_history = False
242
+ # if "dp" in repr(self.agent) and "observation.action_history" in self.cfg.agent.config.policy_cfg.input_shapes:
243
+ if "observation.action_history" in self.cfg.agent.config.policy_cfg.input_shapes:
244
+ self.use_action_history = True
245
+
246
+ # self.eval_replay_storage._new_eval_step(0)
247
+
248
+ # if 'vinn' in repr(self.agent) or 'openloop' in repr(self.agent):
249
+ # with open(self.cfg.expert_dataset, 'rb') as f:
250
+ # if self.cfg.obs_type == 'pixels':
251
+ # self.expert_demo, _, self.expert_action, self.expert_reward = pickle.load(f)
252
+ # elif self.cfg.obs_type == 'features':
253
+ # _, self.expert_demo, self.expert_action, self.expert_reward = pickle.load(f)
254
+
255
+ # if self.cfg.action_trajectories:
256
+ # with open(self.cfg.expert_action_trajectories, 'rb') as f:
257
+ # self.expert_action = pickle.load(f)
258
+
259
+ # if isinstance(self.cfg.train_demo_idxs_list_or_num, int):
260
+ # if self.cfg.train_demo_idxs_list_or_num == -1:
261
+ # self.cfg.train_demo_idxs_list_or_num = len(self.expert_demo)
262
+ # train_demo_idxs_list_or_num = list(range(self.cfg.train_demo_idxs_list_or_num))
263
+
264
+ # self.expert_demo = self.expert_demo[train_demo_idxs_list_or_num]
265
+ # self.expert_action = self.expert_action[train_demo_idxs_list_or_num]
266
+ # self.expert_reward = self.expert_reward[train_demo_idxs_list_or_num]
267
+ # # if self.cfg.action_plans:
268
+ # # self.expert_action_plans = self.expert_action_plans[self.cfg.train_demo_idxs_list_or_num]
269
+ # # self.expert_demo = self.expert_demo[:self.cfg.num_demos]
270
+ # # self.expert_action = self.expert_action[:self.cfg.num_demos]
271
+ # # self.expert_reward = self.expert_reward[:self.cfg.num_demos]
272
+
273
+ # self.expert_demo = np.concatenate(self.expert_demo, axis=0)
274
+ # self.expert_rgb_obs = np.ascontiguousarray(np.transpose(self.expert_demo, (0,2,3,1))[:, :,:,:3].astype(np.uint8))
275
+ # self.expert_action = np.concatenate(self.expert_action, axis=0)
276
+
277
+ # self.agent.save_representations(self.expert_demo, self.expert_action, 128, config=self.expert_demo_config)
278
+
279
+ # Get start points
280
+ if self.cfg.random_start:
281
+ eval_starts = Path(self.cfg.eval_starts) / 'starts.pkl'
282
+ if eval_starts.exists():
283
+ with eval_starts.open('rb') as f:
284
+ self.start_pos = pickle.load(f)
285
+ else:
286
+ eval_starts = Path(self.cfg.eval_starts)
287
+ eval_starts.mkdir(parents=True, exist_ok=True)
288
+
289
+ # Generate start points
290
+ self.start_pos = []
291
+ try:
292
+ for _ in range(self.cfg.num_eval):
293
+ self.start_pos.append(self.eval_env.get_random_pos())
294
+ except:
295
+ for _ in range(self.cfg.num_eval):
296
+ self.start_pos.append(self.eval_env.arm.get_random_pos())
297
+
298
+ # Save start points for the task
299
+ eval_starts = eval_starts / 'starts.pkl'
300
+ with eval_starts.open('wb') as f:
301
+ pickle.dump(self.start_pos, f)
302
+
303
+ time_step = self.eval_env.reset()
304
+ # replay_thread = None
305
+ while eval_until_episode(self.global_episode) and not self.exit_eval:
306
+ # self.video_recorder.init(self.eval_env, video_filename=f'{self.global_episode}_eval.mp4')
307
+ print(f"Starting episode {self.global_episode}")
308
+ time_step = self.eval_env.reset() #Leon: need to call reset twice in case objects are trapped
309
+ self.video_recorder.init(self.eval_env, video_filename=f'{self.global_episode}_eval.mp4')
310
+ # x = input("Press Enter to continue... after reseting env")
311
+ print("Press Enter to continue... after reseting env. To rate prev episode, press 0 for failure and 1 for success")
312
+ self.proceed_after_env_reset_event.clear() # clear the event flag
313
+ self.proceed_after_env_reset_event.wait() # blocking wait for the event flag to be set
314
+ if self.global_episode > 0:
315
+ self.logger.log_metrics({'num_success': self.num_episode_successes}, self.global_step, 'eval', episode=self.global_episode)
316
+ self.logger.log_metrics({'success_rate': self.num_episode_successes/self.global_episode}, self.global_step, 'eval', episode=self.global_episode)
317
+ time_step = self.eval_env.reset()
318
+ # debug_info_dict = self.eval_env.debug_info_dict
319
+ # if replay_thread is not None:
320
+ # # wait for the last replay thread to finish
321
+ # replay_thread.join()
322
+
323
+ # self.eval_replay_storage.add(time_step._replace(observation=time_step.observation[self.cfg.obs_type]), debug_info_dict)
324
+ # replay_thread = threading.Thread(target=self.eval_replay_storage.add, args=(time_step._replace(observation=time_step.observation[self.cfg.obs_type]), debug_info_dict))
325
+ # replay_thread = threading.Thread(target=self.eval_replay_storage.add, args=(time_step, debug_info_dict))
326
+
327
+ # replay_thread.start()
328
+ if self.cfg.random_start:
329
+ time_step = self.reset(self.global_episode)
330
+ time.sleep(2) #5)
331
+ # if 'vinn' in repr(self.agent):
332
+ # self.agent.reset()
333
+ # # self.agent.buffer.reset()
334
+ # # if self.cfg.open_loop:
335
+ # # self.agent.current_step = 0
336
+ # if 'openloop' in repr(self.agent):
337
+ # self.agent.curr_step = 0
338
+ # at start of each episode, provide zero action for policies that use action history
339
+ # shape should be (T_o, T_a, action_dim)
340
+
341
+ # while not time_step.last() and not self.preempt_episode:
342
+ self.video_recorder.ros_start_recording()
343
+ self.eval_replay_storage.start_episode()
344
+ self.eval_env.start_policy_timer()
345
+ while not self.eval_env.episode_done() and not self.preempt_episode:
346
+ # with torch.no_grad(), utils.eval_mode(self.agent):
347
+ # # if self.cfg.agent.provide_topk:
348
+ # # action, vinn_action, topk = self.agent.act(
349
+ # # time_step.observation['pixels'],
350
+ # # self.global_step,
351
+ # # eval_mode=True)
352
+ # # elif self.cfg.agent.provide_obs:
353
+ # # action, vinn_action, obs = self.agent.act(
354
+ # # time_step.observation['pixels'],
355
+ # # self.global_step,
356
+ # # eval_mode=True)
357
+ # # else:
358
+ # action, vinn_action = self.agent.act(
359
+ # time_step.observation,
360
+ # self.global_step,
361
+ # eval_mode=True,
362
+ # obs_timestamp=time_step.observation['timestamp'],
363
+ # obs_seq=time_step.observation['seq'],
364
+ # action_history=action_history,
365
+ # action_history_start_timestamp=action_history_start_timestamp,
366
+ # )
367
+ # DONT WAIT FOR POLICY TO GET AN ACTION
368
+ # we dont want to slow down grabbing obs and passing to sam/contact features
369
+
370
+ self.eval_env.run_policy_threads() # this just does a rospy sleep
371
+
372
+ # if self.use_action_history:
373
+ # action_history_start_timestamp = time_step.observation['timestamp']
374
+ # # action_history = action[:self.cfg.agent.config.policy_cfg.action_history_encoder_config.history_length, ...]
375
+ # # add n_obs_steps dimension to action_history, for now we assume n_obs_steps = 1
376
+ # # TODO: handle n_obs_steps > 1
377
+ # action_history = action[np.newaxis, ...]
378
+
379
+ # time_step = self.eval_env.step(action, vinn_action) # obs, reward after action has been taken
380
+ # debug_info_dict = self.eval_env.debug_info_dict
381
+
382
+ # time_step = self.eval_env.ros_step()
383
+
384
+ # replay_thread.join()
385
+
386
+ # time how long it takes to execute the step
387
+ # time_before_add = time.perf_counter()
388
+ # self.eval_replay_storage.add(time_step._replace(observation=time_step.observation[self.cfg.obs_type]), debug_info_dict)
389
+ # use thread to call the add function in a separate thread
390
+ # replay_thread = threading.Thread(target=self.eval_replay_storage.add, args=(time_step._replace(observation=time_step.observation[self.cfg.obs_type]), debug_info_dict))
391
+
392
+ # replay_thread = threading.Thread(target=self.eval_replay_storage.add, args=(time_step, debug_info_dict))
393
+ # replay_thread.start()
394
+
395
+ # print(f"Time to add to replay buffer: {time.perf_counter() - time_before_add}")
396
+
397
+ # self.video_recorder.record(self.eval_env)
398
+ # self._global_step += 1
399
+
400
+ self.eval_env.stop_policy_timer()
401
+
402
+ if self.restart_episode:
403
+ # means we should delete the current episode and start again
404
+ self.restart_episode = False
405
+ self.eval_replay_storage.reset_current_episode()
406
+ self.video_recorder.reset_current_episode()
407
+
408
+ else:
409
+ self.eval_replay_storage.store_current_episode()
410
+ video_filepath = self.video_recorder.save()
411
+ self.logger.log_video(f"eval/{video_filepath.name.rstrip('.mp4')}", video_filepath, self.global_step)
412
+ self._global_episode += 1
413
+
414
+ self.preempt_episode = False # reset preempt_episode flag
415
+
416
+ # self.video_recorder.save(f'{episode}_eval.mp4')
417
+ # get the video file and convert to video tensor to log
418
+
419
+ self.eval_env.reset()
420
+
421
+ print("Evaluation finished. To wrap up, rate prev episode, press 0 for failure and 1 for success")
422
+ self.proceed_after_env_reset_event.clear() # clear the event flag
423
+ self.proceed_after_env_reset_event.wait() # blocking wait for the event flag to be set
424
+ if self.global_episode > 0:
425
+ # self.logger.log_metrics({'num_success': self.num_episode_successes}, self.global_step, 'eval', episode=self.global_episode)
426
+ self.logger.log_metrics({'num_success': self.num_episode_successes}, self.global_step, 'eval', episode=self.global_episode)
427
+ self.logger.log_metrics({'success_rate': self.num_episode_successes/self.global_episode}, self.global_step, 'eval', episode=self.global_episode)
428
+
429
+ self.continue_keypress_thread = False # will stop the keypress thread
430
+ self.keypress_input_thread.join() # wait for the keypress thread to finish
431
+
432
+ def load_checkpoint_conf(self, snapshot_path):
433
+ config_path = snapshot_path.parent / 'config.yaml'
434
+ if not config_path.exists():
435
+ raise FileNotFoundError(f'No snapshot conf found at {config_path}')
436
+ else:
437
+ # load the omegaconf config
438
+ hydra.core.global_hydra.GlobalHydra.instance().clear()
439
+ hydra.initialize(
440
+ str(_relative_path_between(Path(config_path).absolute().parent, Path(__file__).absolute().parent)),
441
+ )
442
+ cfg = hydra.compose(Path(config_path).stem)
443
+ from deepdiff import DeepDiff
444
+ from omegaconf import open_dict
445
+ diff = DeepDiff(OmegaConf.to_container(cfg), OmegaConf.to_container(self.cfg)) # old, new
446
+ # import re
447
+ overwriteable_keys = [f"root{overwritable_key}" for overwritable_key in ["['use_wandb']", "['path_to_depth_extrinsics']", "['eval']", "['root_dir']", "['wandb_notes']", "['agent']['config']['train_cfg']['use_amp']", "['agent']['config']['compile']", "['agent']['config']['policy_cfg']['num_inference_steps']"]]
448
+ if "values_changed" in diff:
449
+ # top_k_checkpoints, wandb_notes, agent.config.train_cfg.use_amp, save_snapshot_every_epochs_diffusion, check_topk_every_epochs_diffusion, validate_diffusion_on_action_loss_every_epochs, train_eval_diffusion_on_action_loss_every_epochs, validate_every_epochs_diffusion
450
+ # for keys above, overwrite the old config with the new config
451
+ for k, v in diff['values_changed'].items():
452
+ # replace any keys that are under "root['suite']"
453
+ if k in overwriteable_keys or k.startswith("root['suite']"):
454
+ print(f"Found changed key {k} with value {v}. Overwriting old checkpoint config")
455
+ if k == "root['agent']['config']['compile']":
456
+ if diff['values_changed'][k]['new_value']:
457
+ self.loading_uncompiled_checkpoint_with_compile = True
458
+ elif not diff['values_changed'][k]['new_value']:
459
+ # raise ValueError("Cannot load a compiled checkpoint without compile")
460
+ self.loading_compiled_checkpoint_with_no_compile = True
461
+ exec(f"{k.replace('root[', 'cfg[')} = {k.replace('root[', 'self.cfg[')}")
462
+ # for any new values, update the old checkpoint config
463
+ if "dictionary_item_added" in diff:
464
+ for new_key in diff['dictionary_item_added']: # this is a list
465
+ # if new_key == "root['suite']['task_make_fn']['observation_cfg']":
466
+ if new_key == "root['suite']['task_make_fn']['agent_policy_cfg']":
467
+ # pass the agents observation_cfg to the suite task_make_fn
468
+ with open_dict(cfg): # to allow addition of non-existing keys
469
+ # cfg.suite.task_make_fn.observation_cfg = cfg.agent.config.observation_cfg
470
+ cfg.suite.task_make_fn.agent_policy_cfg = cfg.agent.config
471
+ continue
472
+ elif "['agent']['config']['policy_cfg']['input_shapes']" in new_key:
473
+ # skip adding the new key if it is the input_shapes of the policy_cfg
474
+ continue
475
+ else:
476
+ print(f"Found new key {new_key} with value {eval(new_key.replace('root[', 'self.cfg['))}. Adding to checkpoint config")
477
+ # eval(new_key.replace('root', 'cfg')) = eval(new_key.replace('root', 'self.cfg'))
478
+ if new_key == "root['agent']['config']['compile']":
479
+ if self.cfg.agent.config.compile:
480
+ self.loading_uncompiled_checkpoint_with_compile = True
481
+
482
+ with open_dict(cfg):
483
+ exec(f"{new_key.replace('root[', 'cfg[')}={new_key.replace('root[', 'self.cfg[')}")
484
+ self.cfg = cfg
485
+
486
+ def load_checkpoint(self, snapshot_path, bc=False):
487
+ print(f'resuming {repr(self.agent)}: {snapshot_path}')
488
+ with snapshot_path.open('rb') as f:
489
+ payload = torch.load(f)
490
+ agent_payload = {}
491
+ for k, v in payload.items():
492
+ if k not in self.__dict__:
493
+ agent_payload[k] = v
494
+ elif k == '_global_epoch':
495
+ self._global_epoch = v
496
+ print(f'loaded epoch: {v}')
497
+ if self.cfg.use_wandb:
498
+ # add to config of wandb
499
+ wandb.config.update({'epoch': v})
500
+
501
+ # self.agent.load_snapshot_eval(agent_payload, bc)
502
+
503
+ @hydra.main(config_path='cfgs', config_name='config_eval')
504
+ def main(cfg):
505
+ from eval_robot import Workspace as W
506
+ root_dir = Path.cwd()
507
+ workspace = W(cfg)
508
+
509
+ workspace.eval()
510
+
511
+ if __name__ == '__main__':
512
+ main()
205324/wandb/run-20241210_205330-52nddla7/files/config.yaml ADDED
@@ -0,0 +1,966 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ wandb_version: 1
2
+
3
+ root_dir:
4
+ desc: null
5
+ value: /home/leonmkim/fish_leon
6
+ replay_buffer_size:
7
+ desc: null
8
+ value: 150000
9
+ replay_buffer_num_workers:
10
+ desc: null
11
+ value: 2
12
+ nstep:
13
+ desc: null
14
+ value: 3
15
+ batch_size:
16
+ desc: null
17
+ value: 128
18
+ seed:
19
+ desc: null
20
+ value: 0
21
+ dataset_shuffle_seed:
22
+ desc: null
23
+ value: 0
24
+ device:
25
+ desc: null
26
+ value: cuda
27
+ save_video:
28
+ desc: null
29
+ value: true
30
+ save_train_video:
31
+ desc: null
32
+ value: true
33
+ use_tb:
34
+ desc: null
35
+ value: true
36
+ use_wandb:
37
+ desc: null
38
+ value: true
39
+ wandb_run_id:
40
+ desc: null
41
+ value: '265626'
42
+ wandb_notes:
43
+ desc: null
44
+ value: 265626_req_265827
45
+ eval:
46
+ desc: null
47
+ value: true
48
+ process_contact_features:
49
+ desc: null
50
+ value: true
51
+ obs_type:
52
+ desc: null
53
+ value: pixels
54
+ use_color:
55
+ desc: null
56
+ value: true
57
+ use_depth:
58
+ desc: null
59
+ value: true
60
+ use_masks:
61
+ desc: null
62
+ value: true
63
+ mask_list:
64
+ desc: null
65
+ value:
66
+ - EE_obj_mask
67
+ mask_representation:
68
+ desc: null
69
+ value: channels
70
+ crop_hw:
71
+ desc: null
72
+ value:
73
+ - 144
74
+ - 144
75
+ crop_down_offset:
76
+ desc: null
77
+ value: 48
78
+ color_crop_type:
79
+ desc: null
80
+ value: null
81
+ depth_crop_type:
82
+ desc: null
83
+ value: null
84
+ segmask_crop_type:
85
+ desc: null
86
+ value: null
87
+ add_crop_binary_mask:
88
+ desc: null
89
+ value: false
90
+ add_coord_conv_map:
91
+ desc: null
92
+ value: false
93
+ use_context_color:
94
+ desc: null
95
+ value: false
96
+ use_context_depth:
97
+ desc: null
98
+ value: false
99
+ use_context_segmask:
100
+ desc: null
101
+ value: false
102
+ context_color_crop_type:
103
+ desc: null
104
+ value: null
105
+ context_depth_crop_type:
106
+ desc: null
107
+ value: null
108
+ context_segmask_crop_type:
109
+ desc: null
110
+ value: null
111
+ context_add_crop_binary_mask:
112
+ desc: null
113
+ value: false
114
+ context_add_coord_conv_map:
115
+ desc: null
116
+ value: false
117
+ use_contact_map:
118
+ desc: null
119
+ value: false
120
+ use_sdf_maps:
121
+ desc: null
122
+ value: false
123
+ use_normals_maps:
124
+ desc: null
125
+ value: false
126
+ which_objects:
127
+ desc: null
128
+ value: both
129
+ max_contact_prob:
130
+ desc: null
131
+ value: 0.1
132
+ max_depth:
133
+ desc: null
134
+ value: 2.0
135
+ grasped_dtc_max_value:
136
+ desc: null
137
+ value: 0.2
138
+ env_dtc_max_value:
139
+ desc: null
140
+ value: 0.4
141
+ grasped_normals_mask_max_dtc_value:
142
+ desc: null
143
+ value: 0.2
144
+ env_normals_mask_max_dtc_value:
145
+ desc: null
146
+ value: 0.4
147
+ clamp_dtc:
148
+ desc: null
149
+ value: true
150
+ dtc_adaptive_normalization:
151
+ desc: null
152
+ value: false
153
+ mask_normals_within_sdf:
154
+ desc: null
155
+ value: true
156
+ adaptive_normals_mask:
157
+ desc: null
158
+ value: true
159
+ learnable_contact_preprocess_params:
160
+ desc: null
161
+ value: true
162
+ contact_model_name:
163
+ desc: null
164
+ value: local_multitask_outhd64all_home_crop_h144w144d48_mask_ctxtmask_seed_220979_epoch_9
165
+ contact_estimation_model_ckpt_path:
166
+ desc: null
167
+ value: ~/fish_leon/contact_estimation/artifacts/175604_2/checkpoints/epoch=09-val_loss=0.00.ckpt
168
+ encoder_type:
169
+ desc: null
170
+ value: small
171
+ debug_timestamps:
172
+ desc: null
173
+ value: false
174
+ open_loop:
175
+ desc: null
176
+ value: false
177
+ action_trajectories:
178
+ desc: null
179
+ value: true
180
+ stop_after_action:
181
+ desc: null
182
+ value: false
183
+ interpolation_frequency:
184
+ desc: null
185
+ value: 25
186
+ policy_frequency:
187
+ desc: null
188
+ value: 5
189
+ wait_for_new_camera_frames:
190
+ desc: null
191
+ value: true
192
+ baseline:
193
+ desc: null
194
+ value: false
195
+ train_demo_idxs_list_or_num:
196
+ desc: null
197
+ value: -1
198
+ log_train_every_steps:
199
+ desc: null
200
+ value: 25
201
+ name_of_expert_demo:
202
+ desc: null
203
+ value: 64_240x320_hbm_twodim_fps_fix_contact_fix_recovery_fix_annotated_start_idx_5hz_zstd7_EE_pxl_coords_expert_demos_imp_act
204
+ expert_dataset_dirpath:
205
+ desc: null
206
+ value: /home/leonmkim/fish_leon/FISH/expert_demos/frankagym/FrankaInsertion-v1/64_240x320_hbm_twodim_fps_fix_contact_fix_recovery_fix_annotated_start_idx_5hz_zstd7_EE_pxl_coords_expert_demos_imp_act
207
+ store_dataset_in_memory:
208
+ desc: null
209
+ value: false
210
+ expert_dataset:
211
+ desc: null
212
+ value: /home/leonmkim/fish_leon/FISH/expert_demos/frankagym/FrankaInsertion-v1/64_240x320_hbm_twodim_fps_fix_contact_fix_recovery_fix_annotated_start_idx_5hz_zstd7_EE_pxl_coords_expert_demos_imp_act/demos.zarr
213
+ action_key:
214
+ desc: null
215
+ value: action_trajectory_25hz
216
+ semantic_demo_grouping_name:
217
+ desc: null
218
+ value: semantic_demo_grouping.yaml
219
+ semantic_demo_grouping:
220
+ desc: null
221
+ value: /home/leonmkim/fish_leon/FISH/expert_demos/frankagym/FrankaInsertion-v1/64_240x320_hbm_twodim_fps_fix_contact_fix_recovery_fix_annotated_start_idx_5hz_zstd7_EE_pxl_coords_expert_demos_imp_act/semantic_demo_grouping.yaml
222
+ include_groups_list:
223
+ desc: null
224
+ value:
225
+ - hbm_twodim_left_to_right
226
+ expert_dataset_config:
227
+ desc: null
228
+ value: /home/leonmkim/fish_leon/FISH/expert_demos/frankagym/FrankaInsertion-v1/64_240x320_hbm_twodim_fps_fix_contact_fix_recovery_fix_annotated_start_idx_5hz_zstd7_EE_pxl_coords_expert_demos_imp_act/demo_config.yaml
229
+ name_of_valid_demo:
230
+ desc: null
231
+ value: 64_240x320_hbm_twodim_fps_fix_contact_fix_recovery_fix_annotated_start_idx_5hz_zstd7_EE_pxl_coords_expert_demos_imp_act
232
+ valid_dataset_dir:
233
+ desc: null
234
+ value: /home/leonmkim/fish_leon/FISH/expert_demos/frankagym/FrankaInsertion-v1/64_240x320_hbm_twodim_fps_fix_contact_fix_recovery_fix_annotated_start_idx_5hz_zstd7_EE_pxl_coords_expert_demos_imp_act/demos.zarr
235
+ valid_demo_idxs_list_or_num:
236
+ desc: null
237
+ value: null
238
+ val_num_groups:
239
+ desc: null
240
+ value: 0
241
+ load_bc:
242
+ desc: null
243
+ value: true
244
+ checkpoint_epoch_list:
245
+ desc: null
246
+ value:
247
+ - 99
248
+ - 199
249
+ - 299
250
+ - 399
251
+ - 499
252
+ - 599
253
+ - 699
254
+ - 799
255
+ - 899
256
+ - 999
257
+ - 1249
258
+ - 1499
259
+ - 1749
260
+ - 1999
261
+ - 2999
262
+ - 3999
263
+ - 4999
264
+ - 5999
265
+ - 6999
266
+ - 7999
267
+ - 8999
268
+ - 9999
269
+ snapshot_root_dir:
270
+ desc: null
271
+ value: /mnt/grasp_high_usage/leonmkim/contact_estimation/FISH
272
+ save_snapshot:
273
+ desc: null
274
+ value: true
275
+ save_last_snapshot:
276
+ desc: null
277
+ value: true
278
+ save_snapshot_when_done:
279
+ desc: null
280
+ value: true
281
+ top_k_checkpoints:
282
+ desc: null
283
+ value: 5
284
+ save_snapshot_link_to_weights_dir:
285
+ desc: null
286
+ value: deprecated
287
+ bc_regularize:
288
+ desc: null
289
+ value: false
290
+ bc_weight_type:
291
+ desc: null
292
+ value: qfilter
293
+ experiment_dir:
294
+ desc: null
295
+ value: ./exp_local/frankagym_pixels/FrankaInsertion-v1/265626
296
+ agent:
297
+ desc: null
298
+ value:
299
+ _target_: agent.diffusion_policy.DiffusionPolicyAgent
300
+ name: diffusion_policy
301
+ load_checkpoint: true
302
+ obs_shape:
303
+ - 5
304
+ - 240
305
+ - 320
306
+ action_shape:
307
+ - 36
308
+ - 7
309
+ device: cuda
310
+ n_obs_steps: 1
311
+ suite_name: frankagym
312
+ obs_type: pixels
313
+ enable_arm: true
314
+ enable_camera: true
315
+ use_tb: true
316
+ desired_image_shape:
317
+ - 13
318
+ - 180
319
+ - 240
320
+ orig_cam_shape:
321
+ - 3
322
+ - 240
323
+ - 320
324
+ config:
325
+ _target_: agent.diffusion_policy.DiffusionPolicyAgentConfig
326
+ compile: false
327
+ device: cuda
328
+ cam_resize_shape:
329
+ - 13
330
+ - 180
331
+ - 240
332
+ orig_cam_shape:
333
+ - 3
334
+ - 240
335
+ - 320
336
+ policy_cfg:
337
+ _target_: lerobot.common.policies.diffusion.configuration_diffusion.DiffusionConfig
338
+ n_obs_steps: 1
339
+ horizon: 36
340
+ n_action_steps: 36
341
+ output_shapes:
342
+ action:
343
+ - 7
344
+ input_normalization_modes:
345
+ observation.image: mean_std
346
+ observation.state: min_max
347
+ observation.action_history: min_max
348
+ output_normalization_modes:
349
+ action: min_max
350
+ vision_backbone: resnet18
351
+ pretrained_backbone_weights: null
352
+ transforms:
353
+ - _target_: torchaug.transforms.RandomAffine
354
+ degrees:
355
+ - -5
356
+ - 5
357
+ translate:
358
+ - 0.05
359
+ - 0.05
360
+ batch_transform: true
361
+ num_chunks: -1
362
+ batch_inplace: true
363
+ - _target_: torchaug.transforms.RandomColorJitter
364
+ brightness: 0.3
365
+ contrast: 0.4
366
+ saturation: 0.5
367
+ hue: 0.08
368
+ batch_transform: true
369
+ num_chunks: -1
370
+ batch_inplace: true
371
+ use_group_norm: true
372
+ spatial_softmax_num_keypoints: 32
373
+ action_history_encoder_config:
374
+ _target_: lerobot.common.policies.diffusion.configuration_diffusion.Unet1dEncoderConfig
375
+ in_channels: 7
376
+ out_channels: 32
377
+ history_length: 6
378
+ kernel_size: 5
379
+ downsample_kernel_size: 3
380
+ downsample_stride: 2
381
+ downsample_padding: 1
382
+ down_dims:
383
+ - 256
384
+ - 512
385
+ - 1024
386
+ kernel_size: 5
387
+ n_groups: 8
388
+ diffusion_step_embed_dim: 128
389
+ use_film_scale_modulation: true
390
+ noise_scheduler_type: DDIM
391
+ beta_schedule: squaredcos_cap_v2
392
+ beta_start: 0.0001
393
+ beta_end: 0.02
394
+ prediction_type: epsilon
395
+ clip_sample: true
396
+ clip_sample_range: 1.0
397
+ num_train_timesteps: 50
398
+ num_inference_steps: 10
399
+ do_mask_loss_for_padding: false
400
+ input_shapes:
401
+ observation.image:
402
+ - 13
403
+ - 180
404
+ - 240
405
+ context_observation.image:
406
+ - 13
407
+ - 180
408
+ - 240
409
+ observation.state:
410
+ - 8
411
+ observation.action_history:
412
+ - 7
413
+ train_cfg:
414
+ _target_: utils.TrainConfig
415
+ lr: 0.0001
416
+ lr_scheduler: cosine
417
+ lr_warmup_steps: 500
418
+ adam_betas:
419
+ - 0.95
420
+ - 0.999
421
+ adam_eps: 1.0e-08
422
+ adam_weight_decay: 1.0e-06
423
+ grad_clip_norm: 10
424
+ offline_steps: 1000000
425
+ use_amp: true
426
+ observation_cfg:
427
+ _target_: agent.encoder.VisualFeatureSet
428
+ use_depth: true
429
+ use_color: true
430
+ mask_input_dict:
431
+ _target_: agent.encoder.MaskInputDict
432
+ enable: true
433
+ representation: channels
434
+ mask_list:
435
+ - EE_obj_mask
436
+ crop_input_config:
437
+ _target_: agent.encoder.CropInputConfig
438
+ color_crop_type: null
439
+ depth_crop_type: null
440
+ segmask_crop_type: null
441
+ crop_hw:
442
+ - 144
443
+ - 144
444
+ crop_down_offset: 48
445
+ add_crop_binary_mask: false
446
+ add_coord_conv_map: false
447
+ context_input_config:
448
+ _target_: agent.encoder.ContextInputConfig
449
+ use_color: false
450
+ use_depth: false
451
+ mask_input_dict:
452
+ _target_: agent.encoder.MaskInputDict
453
+ enable: false
454
+ representation: channels
455
+ mask_list:
456
+ - EE_obj_mask
457
+ crop_input_config:
458
+ _target_: agent.encoder.CropInputConfig
459
+ color_crop_type: null
460
+ depth_crop_type: null
461
+ segmask_crop_type: null
462
+ crop_hw:
463
+ - 144
464
+ - 144
465
+ crop_down_offset: 48
466
+ add_crop_binary_mask: false
467
+ add_coord_conv_map: false
468
+ mask_soft_approx_scheduler_config:
469
+ _target_: agent.encoder.MaskSoftApproxSchedulerConfig
470
+ num_steps: 40000
471
+ initial_value: 10.0
472
+ final_value: 1000.0
473
+ interpolation_scheme: cosine
474
+ use_contact_map: false
475
+ use_sdf_maps: false
476
+ use_normals_maps: false
477
+ which_objects: both
478
+ grasped_dtc_max_value: 0.2
479
+ env_dtc_max_value: 0.4
480
+ grasped_normals_mask_max_dtc_value: 0.2
481
+ env_normals_mask_max_dtc_value: 0.4
482
+ clamp_dtc: true
483
+ max_contact_prob: 0.1
484
+ mask_normals_within_sdf: true
485
+ dtc_adaptive_normalization: false
486
+ adaptive_normals_mask: true
487
+ max_depth: 2.0
488
+ image_shape:
489
+ - 13
490
+ - 180
491
+ - 240
492
+ learnable_contact_preprocess_params: true
493
+ learning_rate: 0.0001
494
+ weight_decay: 0.0
495
+ contact_model_name: local_multitask_outhd64all_home_crop_h144w144d48_mask_ctxtmask_seed_220979_epoch_9
496
+ zero_centered: false
497
+ policy_frequency: 5
498
+ interpolation_frequency: 25
499
+ suite:
500
+ desc: null
501
+ value:
502
+ suite: frankagym
503
+ name: frankagym
504
+ frame_stack: 1
505
+ action_repeat: 1
506
+ discount: 0.99
507
+ hidden_dim: 1024
508
+ num_train_frames: 2010
509
+ num_seed_frames: 260
510
+ num_train_epochs: 5000
511
+ validate_every_epochs: 100
512
+ validate_diffusion_on_action_loss_every_epochs: 500
513
+ train_eval_diffusion_on_action_loss_every_epochs: 500
514
+ check_topk_every_epochs: 10
515
+ save_snapshot_every_epochs: 5000
516
+ eval_every_frames: 2000
517
+ num_eval_episodes: 5
518
+ save_snapshot: true
519
+ wait_for_user_to_start_episode: true
520
+ task_make_fn:
521
+ _target_: suite.frankagym.make
522
+ name: FrankaInsertion-v1
523
+ height: 240
524
+ width: 320
525
+ frame_stack: 1
526
+ action_repeat: 1
527
+ seed: 0
528
+ enable_arm: true
529
+ enable_gripper: true
530
+ start_with_gripper_open: true
531
+ enable_camera: true
532
+ path_to_depth_extrinsics: ~/fish_leon/FISH/cfgs/camera_poses/camera_poses_L515/20240904-122305/color_tf_world.npy
533
+ contact_estimation_model_ckpt_path: ~/fish_leon/contact_estimation/artifacts/175604_2/checkpoints/epoch=09-val_loss=0.00.ckpt
534
+ x_limit:
535
+ - 0.2
536
+ - 0.7
537
+ y_limit:
538
+ - -0.4
539
+ - 0.4
540
+ z_limit:
541
+ - -0.05
542
+ - 0.55
543
+ device: cuda
544
+ interpolation_frequency: 25
545
+ policy_frequency: 5
546
+ debug_timestamps: false
547
+ stop_after_action: false
548
+ open_loop: false
549
+ wait_for_new_camera_frames: true
550
+ action_key: action_trajectory_25hz
551
+ action_trajectory_horizon: 36
552
+ action_trajectories: true
553
+ path_to_zarr_dataset: /home/leonmkim/fish_leon/FISH/expert_demos/frankagym/FrankaInsertion-v1/64_240x320_hbm_twodim_fps_fix_contact_fix_recovery_fix_annotated_start_idx_5hz_zstd7_EE_pxl_coords_expert_demos_imp_act/demos.zarr
554
+ observation_cfg:
555
+ _target_: agent.encoder.VisualFeatureSet
556
+ use_depth: true
557
+ use_color: true
558
+ mask_input_dict:
559
+ _target_: agent.encoder.MaskInputDict
560
+ enable: true
561
+ representation: channels
562
+ mask_list:
563
+ - EE_obj_mask
564
+ crop_input_config:
565
+ _target_: agent.encoder.CropInputConfig
566
+ color_crop_type: null
567
+ depth_crop_type: null
568
+ segmask_crop_type: null
569
+ crop_hw:
570
+ - 144
571
+ - 144
572
+ crop_down_offset: 48
573
+ add_crop_binary_mask: false
574
+ add_coord_conv_map: false
575
+ context_input_config:
576
+ _target_: agent.encoder.ContextInputConfig
577
+ use_color: false
578
+ use_depth: false
579
+ mask_input_dict:
580
+ _target_: agent.encoder.MaskInputDict
581
+ enable: false
582
+ representation: channels
583
+ mask_list:
584
+ - EE_obj_mask
585
+ crop_input_config:
586
+ _target_: agent.encoder.CropInputConfig
587
+ color_crop_type: null
588
+ depth_crop_type: null
589
+ segmask_crop_type: null
590
+ crop_hw:
591
+ - 144
592
+ - 144
593
+ crop_down_offset: 48
594
+ add_crop_binary_mask: false
595
+ add_coord_conv_map: false
596
+ mask_soft_approx_scheduler_config:
597
+ _target_: agent.encoder.MaskSoftApproxSchedulerConfig
598
+ num_steps: 40000
599
+ initial_value: 10.0
600
+ final_value: 1000.0
601
+ interpolation_scheme: cosine
602
+ use_contact_map: false
603
+ use_sdf_maps: false
604
+ use_normals_maps: false
605
+ which_objects: both
606
+ grasped_dtc_max_value: 0.2
607
+ env_dtc_max_value: 0.4
608
+ grasped_normals_mask_max_dtc_value: 0.2
609
+ env_normals_mask_max_dtc_value: 0.4
610
+ clamp_dtc: true
611
+ max_contact_prob: 0.1
612
+ mask_normals_within_sdf: true
613
+ dtc_adaptive_normalization: false
614
+ adaptive_normals_mask: true
615
+ max_depth: 2.0
616
+ image_shape:
617
+ - 13
618
+ - 180
619
+ - 240
620
+ learnable_contact_preprocess_params: true
621
+ learning_rate: 0.0001
622
+ weight_decay: 0.0
623
+ contact_model_name: local_multitask_outhd64all_home_crop_h144w144d48_mask_ctxtmask_seed_220979_epoch_9
624
+ zero_centered: false
625
+ agent_policy_cfg:
626
+ _target_: agent.diffusion_policy.DiffusionPolicyAgentConfig
627
+ compile: false
628
+ device: cuda
629
+ cam_resize_shape:
630
+ - 13
631
+ - 180
632
+ - 240
633
+ orig_cam_shape:
634
+ - 3
635
+ - 240
636
+ - 320
637
+ policy_cfg:
638
+ _target_: lerobot.common.policies.diffusion.configuration_diffusion.DiffusionConfig
639
+ n_obs_steps: 1
640
+ horizon: 36
641
+ n_action_steps: 36
642
+ output_shapes:
643
+ action:
644
+ - 7
645
+ input_normalization_modes:
646
+ observation.image: mean_std
647
+ observation.state: min_max
648
+ observation.action_history: min_max
649
+ output_normalization_modes:
650
+ action: min_max
651
+ vision_backbone: resnet18
652
+ pretrained_backbone_weights: null
653
+ transforms:
654
+ - _target_: torchaug.transforms.RandomAffine
655
+ degrees:
656
+ - -5
657
+ - 5
658
+ translate:
659
+ - 0.05
660
+ - 0.05
661
+ batch_transform: true
662
+ num_chunks: -1
663
+ batch_inplace: true
664
+ - _target_: torchaug.transforms.RandomColorJitter
665
+ brightness: 0.3
666
+ contrast: 0.4
667
+ saturation: 0.5
668
+ hue: 0.08
669
+ batch_transform: true
670
+ num_chunks: -1
671
+ batch_inplace: true
672
+ use_group_norm: true
673
+ spatial_softmax_num_keypoints: 32
674
+ action_history_encoder_config:
675
+ _target_: lerobot.common.policies.diffusion.configuration_diffusion.Unet1dEncoderConfig
676
+ in_channels: 7
677
+ out_channels: 32
678
+ history_length: 6
679
+ kernel_size: 5
680
+ downsample_kernel_size: 3
681
+ downsample_stride: 2
682
+ downsample_padding: 1
683
+ down_dims:
684
+ - 256
685
+ - 512
686
+ - 1024
687
+ kernel_size: 5
688
+ n_groups: 8
689
+ diffusion_step_embed_dim: 128
690
+ use_film_scale_modulation: true
691
+ noise_scheduler_type: DDIM
692
+ beta_schedule: squaredcos_cap_v2
693
+ beta_start: 0.0001
694
+ beta_end: 0.02
695
+ prediction_type: epsilon
696
+ clip_sample: true
697
+ clip_sample_range: 1.0
698
+ num_train_timesteps: 50
699
+ num_inference_steps: 10
700
+ do_mask_loss_for_padding: false
701
+ input_shapes:
702
+ observation.image:
703
+ - 13
704
+ - 180
705
+ - 240
706
+ context_observation.image:
707
+ - 13
708
+ - 180
709
+ - 240
710
+ observation.state:
711
+ - 8
712
+ observation.action_history:
713
+ - 7
714
+ train_cfg:
715
+ _target_: utils.TrainConfig
716
+ lr: 0.0001
717
+ lr_scheduler: cosine
718
+ lr_warmup_steps: 500
719
+ adam_betas:
720
+ - 0.95
721
+ - 0.999
722
+ adam_eps: 1.0e-08
723
+ adam_weight_decay: 1.0e-06
724
+ grad_clip_norm: 10
725
+ offline_steps: 1000000
726
+ use_amp: true
727
+ observation_cfg:
728
+ _target_: agent.encoder.VisualFeatureSet
729
+ use_depth: true
730
+ use_color: true
731
+ mask_input_dict:
732
+ _target_: agent.encoder.MaskInputDict
733
+ enable: true
734
+ representation: channels
735
+ mask_list:
736
+ - EE_obj_mask
737
+ crop_input_config:
738
+ _target_: agent.encoder.CropInputConfig
739
+ color_crop_type: null
740
+ depth_crop_type: null
741
+ segmask_crop_type: null
742
+ crop_hw:
743
+ - 144
744
+ - 144
745
+ crop_down_offset: 48
746
+ add_crop_binary_mask: false
747
+ add_coord_conv_map: false
748
+ context_input_config:
749
+ _target_: agent.encoder.ContextInputConfig
750
+ use_color: false
751
+ use_depth: false
752
+ mask_input_dict:
753
+ _target_: agent.encoder.MaskInputDict
754
+ enable: false
755
+ representation: channels
756
+ mask_list:
757
+ - EE_obj_mask
758
+ crop_input_config:
759
+ _target_: agent.encoder.CropInputConfig
760
+ color_crop_type: null
761
+ depth_crop_type: null
762
+ segmask_crop_type: null
763
+ crop_hw:
764
+ - 144
765
+ - 144
766
+ crop_down_offset: 48
767
+ add_crop_binary_mask: false
768
+ add_coord_conv_map: false
769
+ mask_soft_approx_scheduler_config:
770
+ _target_: agent.encoder.MaskSoftApproxSchedulerConfig
771
+ num_steps: 40000
772
+ initial_value: 10.0
773
+ final_value: 1000.0
774
+ interpolation_scheme: cosine
775
+ use_contact_map: false
776
+ use_sdf_maps: false
777
+ use_normals_maps: false
778
+ which_objects: both
779
+ grasped_dtc_max_value: 0.2
780
+ env_dtc_max_value: 0.4
781
+ grasped_normals_mask_max_dtc_value: 0.2
782
+ env_normals_mask_max_dtc_value: 0.4
783
+ clamp_dtc: true
784
+ max_contact_prob: 0.1
785
+ mask_normals_within_sdf: true
786
+ dtc_adaptive_normalization: false
787
+ adaptive_normals_mask: true
788
+ max_depth: 2.0
789
+ image_shape:
790
+ - 13
791
+ - 180
792
+ - 240
793
+ learnable_contact_preprocess_params: true
794
+ learning_rate: 0.0001
795
+ weight_decay: 0.0
796
+ contact_model_name: local_multitask_outhd64all_home_crop_h144w144d48_mask_ctxtmask_seed_220979_epoch_9
797
+ zero_centered: false
798
+ policy_frequency: 5
799
+ interpolation_frequency: 25
800
+ true_action_history: false
801
+ num_train_frames_bc:
802
+ desc: null
803
+ value: 50000
804
+ num_train_frames_drq:
805
+ desc: null
806
+ value: 1100000
807
+ stddev_schedule_drq:
808
+ desc: null
809
+ value: linear(1.0,0.1,100000)
810
+ task_name:
811
+ desc: null
812
+ value: FrankaInsertion-v1
813
+ num_train_frames_vinn:
814
+ desc: null
815
+ value: 25000
816
+ num_train_frames_diffusion:
817
+ desc: null
818
+ value: 1000000
819
+ num_train_epochs_bc:
820
+ desc: null
821
+ value: 5000
822
+ num_train_epochs_diffusion:
823
+ desc: null
824
+ value: 15000
825
+ validate_every_epochs_bc:
826
+ desc: null
827
+ value: 5
828
+ validate_every_epochs_diffusion:
829
+ desc: null
830
+ value: 250
831
+ validate_diffusion_on_action_loss_every_epochs:
832
+ desc: null
833
+ value: 250
834
+ train_eval_diffusion_on_action_loss_every_epochs:
835
+ desc: null
836
+ value: 250
837
+ check_topk_every_epochs:
838
+ desc: null
839
+ value: 5
840
+ check_topk_every_epochs_diffusion:
841
+ desc: null
842
+ value: 250
843
+ save_snapshot_every_epochs_diffusion:
844
+ desc: null
845
+ value: 1500
846
+ x_limit:
847
+ desc: null
848
+ value:
849
+ - 0.2
850
+ - 0.7
851
+ y_limit:
852
+ desc: null
853
+ value:
854
+ - -0.4
855
+ - 0.4
856
+ z_limit:
857
+ desc: null
858
+ value:
859
+ - -0.05
860
+ - 0.55
861
+ home_displacement:
862
+ desc: null
863
+ value:
864
+ - 0.55
865
+ - 0.0
866
+ - 0.55
867
+ - 180.0
868
+ - 0.0
869
+ - 0.0
870
+ enable_gripper:
871
+ desc: null
872
+ value: true
873
+ start_with_gripper_open:
874
+ desc: null
875
+ value: true
876
+ offset_mask:
877
+ desc: null
878
+ value:
879
+ - 1
880
+ - 1
881
+ - 1
882
+ - 1
883
+ - 1
884
+ - 1
885
+ path_to_depth_extrinsics:
886
+ desc: null
887
+ value: ~/fish_leon/FISH/cfgs/camera_poses/camera_poses_L515/20240904-122305/color_tf_world.npy
888
+ feature_type:
889
+ desc: null
890
+ value: 180x240_1_RGB_D_2.0_msk_channels_EE_obj_mask_acthst_hst6_out32_dwnkrnl3_dwnstrd2_dwnpd1
891
+ save_buffer:
892
+ desc: null
893
+ value: true
894
+ num_eval:
895
+ desc: null
896
+ value: 5
897
+ random_start:
898
+ desc: null
899
+ value: false
900
+ eval_starts:
901
+ desc: null
902
+ value: /home/leonmkim/fish_leon/FISH/eval_starts/frankagym_pixels/FrankaInsertion-v1
903
+ num_valid_demos:
904
+ desc: null
905
+ value: null
906
+ load_checkpoint:
907
+ desc: null
908
+ value: true
909
+ true_action_history:
910
+ desc: null
911
+ value: false
912
+ checkpoint_epoch:
913
+ desc: null
914
+ value: 12000
915
+ load_residual_weight:
916
+ desc: null
917
+ value: false
918
+ checkpoint_root_dir:
919
+ desc: null
920
+ value: /home/leonmkim/fish_leon/FISH
921
+ checkpoint_weight_dir:
922
+ desc: null
923
+ value: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626
924
+ residual_weight:
925
+ desc: null
926
+ value: /home/leonmkim/fish_leon/FISH/weights/frankagym_pixels/FrankaInsertion-v1/weight.pt
927
+ _wandb:
928
+ desc: null
929
+ value:
930
+ code_path: code/FISH/eval_robot.py
931
+ python_version: 3.10.14
932
+ cli_version: 0.17.5
933
+ framework: torch
934
+ is_jupyter_run: false
935
+ is_kaggle_kernel: false
936
+ start_time: 1733882010
937
+ t:
938
+ 1:
939
+ - 1
940
+ - 41
941
+ - 49
942
+ - 50
943
+ - 55
944
+ - 83
945
+ 2:
946
+ - 1
947
+ - 41
948
+ - 49
949
+ - 50
950
+ - 55
951
+ - 83
952
+ 3:
953
+ - 16
954
+ - 23
955
+ - 35
956
+ 4: 3.10.14
957
+ 5: 0.17.5
958
+ 8:
959
+ - 5
960
+ 13: linux-x86_64
961
+ grasped_obj_name:
962
+ desc: null
963
+ value: hbm
964
+ left_book_slot:
965
+ desc: null
966
+ value: twodim
205324/wandb/run-20241210_205330-52nddla7/files/diff.patch ADDED
@@ -0,0 +1,34 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ diff --git a/FISH/cfgs/config_eval.yaml b/FISH/cfgs/config_eval.yaml
2
+ index a9568e3..aa52de5 100644
3
+ --- a/FISH/cfgs/config_eval.yaml
4
+ +++ b/FISH/cfgs/config_eval.yaml
5
+ @@ -123,6 +123,11 @@ bc_weight_type: 'qfilter' # linear, qfilter
6
+
7
+ # Load weights
8
+ load_checkpoint: ${agent.load_checkpoint}
9
+ +# 64_240x320_hbm_twodim_fps_fix_contact_fix_recovery_fix_annotated_start_idx_5hz_zstd7_EE_pxl_coords_expert_demos_imp_act
10
+ +# full dataset
11
+ +wandb_run_id: "265626" # RGBD+Mask+act history
12
+ +# wandb_run_id: "265628" # RGBD+Mask+contact+act history
13
+ +
14
+ # 54_240x320_hbm_twodim_fps_fix_annotated_start_idx_5hz_zstd7_EE_pxl_coords_expert_demos_imp_act
15
+ # full dataset
16
+ # 6 step action history
17
+ @@ -135,7 +140,7 @@ load_checkpoint: ${agent.load_checkpoint}
18
+ # wandb_run_id: '264772_0' # RGBD+act history
19
+ # wandb_run_id: '264775_0' # RGBD+mask+act history
20
+ # wandb_run_id: '264776_0' # RGBD+contact+act history
21
+ -wandb_run_id: '264777_0' # RGBD+mask+contact+act history
22
+ +# wandb_run_id: '264777_0' # RGBD+mask+contact+act history
23
+
24
+ # 54_240x320_hbm_twodim_fps_fix_annotated_start_idx_5hz_zstd7_EE_pxl_coords_expert_demos_imp_act
25
+ # 16 held out demos
26
+ @@ -151,7 +156,7 @@ wandb_run_id: '264777_0' # RGBD+mask+contact+act history
27
+ # wandb_run_id: '250380_0' # RGBD+mask+contact+act history
28
+ # wandb_run_id: '250370_0' # RGBD+contact+act history
29
+
30
+ -true_action_history: true
31
+ +true_action_history: false
32
+ wandb_notes: null
33
+
34
+ checkpoint_epoch: 12000 # for depth + all contact
205324/wandb/run-20241210_205330-52nddla7/files/output.log ADDED
@@ -0,0 +1,461 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+
2
+ loaded agent with feature_type: 180x240_1_RGB_D_2.0_msk_channels_EE_obj_mask_acthst_hst6_out32_dwnkrnl3_dwnstrd2_dwnpd1
3
+ [INFO] [1733882015.547602]: resetting environment
4
+ [INFO] [1733882015.555685]: cleared current plan
5
+ [INFO] [1733882015.557298]: moving to home
6
+ [INFO] [1733882018.360980]: reached home
7
+ [INFO] [1733882018.362525]: reset action history
8
+ [INFO] [1733882021.071264]: environment reset
9
+ Starting episode 0
10
+ [INFO] [1733882021.073619]: resetting environment
11
+ [INFO] [1733882021.078303]: cleared current plan
12
+ [INFO] [1733882021.079103]: moving to home
13
+ [INFO] [1733882022.081897]: reached home
14
+ [INFO] [1733882022.083956]: reset action history
15
+ [INFO] [1733882024.791365]: environment reset
16
+ Press Enter to continue... after reseting env. To rate prev episode, press 0 for failure and 1 for success
17
+ proceeding to start episode!
18
+ [INFO] [1733882047.498704]: resetting environment
19
+ [INFO] [1733882047.505692]: cleared current plan
20
+ [INFO] [1733882047.506996]: moving to home
21
+ [INFO] [1733882048.509273]: reached home
22
+ [INFO] [1733882048.510630]: reset action history
23
+ [INFO] [1733882051.218254]: environment reset
24
+ bag_path_name:=/home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205324/episode_rosbags/episode_0_2024-12-10-20-54-11.bag
25
+ ... logging to /home/leonmkim/.ros/log/1408e5ba-b74f-11ef-9a34-5546d6409a19/roslaunch-leonmkim-ROG-Strix-G15CS-G15CS-597884.log
26
+ started roslaunch server http://158.130.50.37:45489/
27
+ SUMMARY
28
+ ========
29
+ PARAMETERS
30
+ * /rosdistro: noetic
31
+ * /rosversion: 1.16.0
32
+ NODES
33
+ /
34
+ print_text (rostopic/rostopic)
35
+ pub_text (rostopic/rostopic)
36
+ rosbag_record (rosbag/record)
37
+ ROS_MASTER_URI=http://localhost:11311
38
+ process[pub_text-1]: started with pid [598110]
39
+ process[print_text-2]: started with pid [598134]
40
+ process[rosbag_record-3]: started with pid [598135]
41
+ started bagging!
42
+ [INFO] [1733882051.973240]: received mask_obs with header seq: 110
43
+ stamp:
44
+ secs: 1733882051
45
+ nsecs: 972197771
46
+ frame_id: '' and grasped_object_mask header seq: 11387
47
+ stamp:
48
+ secs: 1733882051
49
+ nsecs: 753522158
50
+ frame_id: "camera_color_optical_frame"
51
+ [INFO] [1733882052.158201]: received mask_obs with header seq: 111
52
+ stamp:
53
+ secs: 1733882052
54
+ nsecs: 157648563
55
+ frame_id: '' and grasped_object_mask header seq: 11393
56
+ stamp:
57
+ secs: 1733882051
58
+ nsecs: 953104734
59
+ frame_id: "camera_color_optical_frame"
60
+ Exception in thread Thread-71:
61
+ Traceback (most recent call last):
62
+ File "/home/leonmkim/.pyenv/versions/3.10.14/lib/python3.10/threading.py", line 1016, in _bootstrap_inner
63
+ self.run()
64
+ File "/opt/ros/noetic/lib/python3/dist-packages/rospy/timer.py", line 237, in run
65
+ self._callback(TimerEvent(last_expected, last_real, current_expected, current_real, last_duration))
66
+ File "/home/leonmkim/fish_leon/gym-envs/gym_envs/envs/franka_env.py", line 562, in ros_step_cb
67
+ action_history, plan_headers_list = self.arm.get_action_history(
68
+ File "/home/leonmkim/fish_leon/gym-envs/gym_envs/envs/hardware/franka.py", line 574, in get_action_history
69
+ action_history, plan_headers_list = self.get_hallucinated_action_history(prev_plan_timestamp, current_timestamp, history_length)
70
+ File "/home/leonmkim/fish_leon/gym-envs/gym_envs/envs/hardware/franka.py", line 599, in get_hallucinated_action_history
71
+ assert action_plan_return is not None, f'No plan found after {timeout.to_sec()} seconds'
72
+ AssertionError: No plan found after 1.0 seconds
73
+ [WARN] [1733882053.167214]: Plan exhausted
74
+ [WARN] [1733882053.207056]: Plan exhausted
75
+ [WARN] [1733882053.247010]: Plan exhausted
76
+ [WARN] [1733882053.287040]: Plan exhausted
77
+ [WARN] [1733882053.327405]: Plan exhausted
78
+ [WARN] [1733882053.367146]: Plan exhausted
79
+ [WARN] [1733882053.407281]: Plan exhausted
80
+ [WARN] [1733882053.447035]: Plan exhausted
81
+ [WARN] [1733882053.487113]: Plan exhausted
82
+ [WARN] [1733882053.527055]: Plan exhausted
83
+ [WARN] [1733882053.567043]: Plan exhausted
84
+ [WARN] [1733882053.607055]: Plan exhausted
85
+ [WARN] [1733882053.647050]: Plan exhausted
86
+ [WARN] [1733882053.687051]: Plan exhausted
87
+ [WARN] [1733882053.727048]: Plan exhausted
88
+ [WARN] [1733882053.767034]: Plan exhausted
89
+ [WARN] [1733882053.807052]: Plan exhausted
90
+ [WARN] [1733882053.847048]: Plan exhausted
91
+ [WARN] [1733882053.887264]: Plan exhausted
92
+ [WARN] [1733882053.927126]: Plan exhausted
93
+ [WARN] [1733882053.967045]: Plan exhausted
94
+ [WARN] [1733882054.007052]: Plan exhausted
95
+ [WARN] [1733882054.047050]: Plan exhausted
96
+ [WARN] [1733882054.087042]: Plan exhausted
97
+ [WARN] [1733882054.127052]: Plan exhausted
98
+ [WARN] [1733882054.167037]: Plan exhausted
99
+ [WARN] [1733882054.207051]: Plan exhausted
100
+ [WARN] [1733882054.247047]: Plan exhausted
101
+ [WARN] [1733882054.287087]: Plan exhausted
102
+ [WARN] [1733882054.327114]: Plan exhausted
103
+ [WARN] [1733882054.367050]: Plan exhausted
104
+ [WARN] [1733882054.407066]: Plan exhausted
105
+ [WARN] [1733882054.447061]: Plan exhausted
106
+ [WARN] [1733882054.487144]: Plan exhausted
107
+ [WARN] [1733882054.527057]: Plan exhausted
108
+ [WARN] [1733882054.567049]: Plan exhausted
109
+ [WARN] [1733882054.607058]: Plan exhausted
110
+ [WARN] [1733882054.647049]: Plan exhausted
111
+ [WARN] [1733882054.687141]: Plan exhausted
112
+ [WARN] [1733882054.727052]: Plan exhausted
113
+ [WARN] [1733882054.767101]: Plan exhausted
114
+ [WARN] [1733882054.807044]: Plan exhausted
115
+ [WARN] [1733882054.847019]: Plan exhausted
116
+ [WARN] [1733882054.887033]: Plan exhausted
117
+ [WARN] [1733882054.927117]: Plan exhausted
118
+ [WARN] [1733882054.967021]: Plan exhausted
119
+ [WARN] [1733882055.007079]: Plan exhausted
120
+ [WARN] [1733882055.047052]: Plan exhausted
121
+ [WARN] [1733882055.087068]: Plan exhausted
122
+ [WARN] [1733882055.127060]: Plan exhausted
123
+ [WARN] [1733882055.167279]: Plan exhausted
124
+ [WARN] [1733882055.207070]: Plan exhausted
125
+ [WARN] [1733882055.247038]: Plan exhausted
126
+ [WARN] [1733882055.287061]: Plan exhausted
127
+ [WARN] [1733882055.327050]: Plan exhausted
128
+ [WARN] [1733882055.367044]: Plan exhausted
129
+ [WARN] [1733882055.407220]: Plan exhausted
130
+ [WARN] [1733882055.447047]: Plan exhausted
131
+ [WARN] [1733882055.487638]: Plan exhausted
132
+ [WARN] [1733882055.527050]: Plan exhausted
133
+ [WARN] [1733882055.567115]: Plan exhausted
134
+ [WARN] [1733882055.607057]: Plan exhausted
135
+ [WARN] [1733882055.647342]: Plan exhausted
136
+ [WARN] [1733882055.687266]: Plan exhausted
137
+ [WARN] [1733882055.727048]: Plan exhausted
138
+ [WARN] [1733882055.767036]: Plan exhausted
139
+ [WARN] [1733882055.807041]: Plan exhausted
140
+ [WARN] [1733882055.847427]: Plan exhausted
141
+ [WARN] [1733882055.887833]: Plan exhausted
142
+ [WARN] [1733882055.927052]: Plan exhausted
143
+ [WARN] [1733882055.967049]: Plan exhausted
144
+ [WARN] [1733882056.007052]: Plan exhausted
145
+ [WARN] [1733882056.047083]: Plan exhausted
146
+ [WARN] [1733882056.087589]: Plan exhausted
147
+ [WARN] [1733882056.127044]: Plan exhausted
148
+ [WARN] [1733882056.167048]: Plan exhausted
149
+ [WARN] [1733882056.207262]: Plan exhausted
150
+ [WARN] [1733882056.247047]: Plan exhausted
151
+ [WARN] [1733882056.287124]: Plan exhausted
152
+ [WARN] [1733882056.327050]: Plan exhausted
153
+ [WARN] [1733882056.367157]: Plan exhausted
154
+ [WARN] [1733882056.407021]: Plan exhausted
155
+ [WARN] [1733882056.447050]: Plan exhausted
156
+ [WARN] [1733882056.487767]: Plan exhausted
157
+ [WARN] [1733882056.527117]: Plan exhausted
158
+ [WARN] [1733882056.567130]: Plan exhausted
159
+ [WARN] [1733882056.607051]: Plan exhausted
160
+ [WARN] [1733882056.647138]: Plan exhausted
161
+ [WARN] [1733882056.687124]: Plan exhausted
162
+ [WARN] [1733882056.727053]: Plan exhausted
163
+ [WARN] [1733882056.767150]: Plan exhausted
164
+ [WARN] [1733882056.807051]: Plan exhausted
165
+ [WARN] [1733882056.847173]: Plan exhausted
166
+ [WARN] [1733882056.887062]: Plan exhausted
167
+ [WARN] [1733882056.927035]: Plan exhausted
168
+ [WARN] [1733882056.967050]: Plan exhausted
169
+ [WARN] [1733882057.007113]: Plan exhausted
170
+ [WARN] [1733882057.047042]: Plan exhausted
171
+ [WARN] [1733882057.087055]: Plan exhausted
172
+ [WARN] [1733882057.127055]: Plan exhausted
173
+ [WARN] [1733882057.167049]: Plan exhausted
174
+ [WARN] [1733882057.207051]: Plan exhausted
175
+ [WARN] [1733882057.247331]: Plan exhausted
176
+ [WARN] [1733882057.287073]: Plan exhausted
177
+ [WARN] [1733882057.327050]: Plan exhausted
178
+ [WARN] [1733882057.367083]: Plan exhausted
179
+ [WARN] [1733882057.407049]: Plan exhausted
180
+ [WARN] [1733882057.447057]: Plan exhausted
181
+ [WARN] [1733882057.487026]: Plan exhausted
182
+ [WARN] [1733882057.527054]: Plan exhausted
183
+ [WARN] [1733882057.567045]: Plan exhausted
184
+ [WARN] [1733882057.607049]: Plan exhausted
185
+ [WARN] [1733882057.647047]: Plan exhausted
186
+ [WARN] [1733882057.687011]: Plan exhausted
187
+ [WARN] [1733882057.727017]: Plan exhausted
188
+ [WARN] [1733882057.767211]: Plan exhausted
189
+ [WARN] [1733882057.807042]: Plan exhausted
190
+ [WARN] [1733882057.847021]: Plan exhausted
191
+ [WARN] [1733882057.887207]: Plan exhausted
192
+ [WARN] [1733882057.927228]: Plan exhausted
193
+ [WARN] [1733882057.967051]: Plan exhausted
194
+ [WARN] [1733882058.007057]: Plan exhausted
195
+ [WARN] [1733882058.047322]: Plan exhausted
196
+ [WARN] [1733882058.087226]: Plan exhausted
197
+ [WARN] [1733882058.127048]: Plan exhausted
198
+ [WARN] [1733882058.167130]: Plan exhausted
199
+ [WARN] [1733882058.207052]: Plan exhausted
200
+ [WARN] [1733882058.247214]: Plan exhausted
201
+ [WARN] [1733882058.287363]: Plan exhausted
202
+ [WARN] [1733882058.327049]: Plan exhausted
203
+ [WARN] [1733882058.367048]: Plan exhausted
204
+ [WARN] [1733882058.407050]: Plan exhausted
205
+ [WARN] [1733882058.447052]: Plan exhausted
206
+ [WARN] [1733882058.487058]: Plan exhausted
207
+ [WARN] [1733882058.527298]: Plan exhausted
208
+ [WARN] [1733882058.567122]: Plan exhausted
209
+ [WARN] [1733882058.607055]: Plan exhausted
210
+ [WARN] [1733882058.647049]: Plan exhausted
211
+ [WARN] [1733882058.687101]: Plan exhausted
212
+ [WARN] [1733882058.727047]: Plan exhausted
213
+ [WARN] [1733882058.767135]: Plan exhausted
214
+ [WARN] [1733882058.807058]: Plan exhausted
215
+ [WARN] [1733882058.847335]: Plan exhausted
216
+ [WARN] [1733882058.887022]: Plan exhausted
217
+ [WARN] [1733882058.927045]: Plan exhausted
218
+ [WARN] [1733882058.967049]: Plan exhausted
219
+ [WARN] [1733882059.007245]: Plan exhausted
220
+ [WARN] [1733882059.047005]: Plan exhausted
221
+ [WARN] [1733882059.087120]: Plan exhausted
222
+ [WARN] [1733882059.127012]: Plan exhausted
223
+ [WARN] [1733882059.167065]: Plan exhausted
224
+ [WARN] [1733882059.207019]: Plan exhausted
225
+ [WARN] [1733882059.247036]: Plan exhausted
226
+ [WARN] [1733882059.287063]: Plan exhausted
227
+ [WARN] [1733882059.327055]: Plan exhausted
228
+ [WARN] [1733882059.367099]: Plan exhausted
229
+ [WARN] [1733882059.407047]: Plan exhausted
230
+ [WARN] [1733882059.447119]: Plan exhausted
231
+ [WARN] [1733882059.487020]: Plan exhausted
232
+ [WARN] [1733882059.527021]: Plan exhausted
233
+ [WARN] [1733882059.567276]: Plan exhausted
234
+ [WARN] [1733882059.607046]: Plan exhausted
235
+ [WARN] [1733882059.647049]: Plan exhausted
236
+ [WARN] [1733882059.687052]: Plan exhausted
237
+ [WARN] [1733882059.727055]: Plan exhausted
238
+ [WARN] [1733882059.767051]: Plan exhausted
239
+ [WARN] [1733882059.807057]: Plan exhausted
240
+ [WARN] [1733882059.847101]: Plan exhausted
241
+ [WARN] [1733882059.887052]: Plan exhausted
242
+ [WARN] [1733882059.927545]: Plan exhausted
243
+ [WARN] [1733882059.967116]: Plan exhausted
244
+ [WARN] [1733882060.007119]: Plan exhausted
245
+ [WARN] [1733882060.047050]: Plan exhausted
246
+ [WARN] [1733882060.087050]: Plan exhausted
247
+ [WARN] [1733882060.127049]: Plan exhausted
248
+ [WARN] [1733882060.167005]: Plan exhausted
249
+ [WARN] [1733882060.207313]: Plan exhausted
250
+ [WARN] [1733882060.247017]: Plan exhausted
251
+ [WARN] [1733882060.287113]: Plan exhausted
252
+ [WARN] [1733882060.327050]: Plan exhausted
253
+ [WARN] [1733882060.367122]: Plan exhausted
254
+ [WARN] [1733882060.407054]: Plan exhausted
255
+ [WARN] [1733882060.447045]: Plan exhausted
256
+ [WARN] [1733882060.487054]: Plan exhausted
257
+ [WARN] [1733882060.527052]: Plan exhausted
258
+ [WARN] [1733882060.567052]: Plan exhausted
259
+ [WARN] [1733882060.607050]: Plan exhausted
260
+ [WARN] [1733882060.647051]: Plan exhausted
261
+ [WARN] [1733882060.687018]: Plan exhausted
262
+ [WARN] [1733882060.727043]: Plan exhausted
263
+ [WARN] [1733882060.767119]: Plan exhausted
264
+ [WARN] [1733882060.807206]: Plan exhausted
265
+ [WARN] [1733882060.847048]: Plan exhausted
266
+ [WARN] [1733882060.887264]: Plan exhausted
267
+ [WARN] [1733882060.927234]: Plan exhausted
268
+ [WARN] [1733882060.967058]: Plan exhausted
269
+ [WARN] [1733882061.007091]: Plan exhausted
270
+ [WARN] [1733882061.047144]: Plan exhausted
271
+ [WARN] [1733882061.087239]: Plan exhausted
272
+ [WARN] [1733882061.127246]: Plan exhausted
273
+ [WARN] [1733882061.167158]: Plan exhausted
274
+ [WARN] [1733882061.207054]: Plan exhausted
275
+ [WARN] [1733882061.247165]: Plan exhausted
276
+ [WARN] [1733882061.287024]: Plan exhausted
277
+ [WARN] [1733882061.327046]: Plan exhausted
278
+ [WARN] [1733882061.367183]: Plan exhausted
279
+ [WARN] [1733882061.407129]: Plan exhausted
280
+ [WARN] [1733882061.447046]: Plan exhausted
281
+ [WARN] [1733882061.487052]: Plan exhausted
282
+ [WARN] [1733882061.527047]: Plan exhausted
283
+ [WARN] [1733882061.567054]: Plan exhausted
284
+ [WARN] [1733882061.607161]: Plan exhausted
285
+ [WARN] [1733882061.647009]: Plan exhausted
286
+ [WARN] [1733882061.687433]: Plan exhausted
287
+ [WARN] [1733882061.727046]: Plan exhausted
288
+ [WARN] [1733882061.767049]: Plan exhausted
289
+ [WARN] [1733882061.807220]: Plan exhausted
290
+ [WARN] [1733882061.847046]: Plan exhausted
291
+ [WARN] [1733882061.887051]: Plan exhausted
292
+ [WARN] [1733882061.927050]: Plan exhausted
293
+ [WARN] [1733882061.967053]: Plan exhausted
294
+ [WARN] [1733882062.007053]: Plan exhausted
295
+ [WARN] [1733882062.047162]: Plan exhausted
296
+ [WARN] [1733882062.087059]: Plan exhausted
297
+ [WARN] [1733882062.127038]: Plan exhausted
298
+ [WARN] [1733882062.167129]: Plan exhausted
299
+ [WARN] [1733882062.207045]: Plan exhausted
300
+ [WARN] [1733882062.247299]: Plan exhausted
301
+ [WARN] [1733882062.287050]: Plan exhausted
302
+ [WARN] [1733882062.327052]: Plan exhausted
303
+ [WARN] [1733882062.367054]: Plan exhausted
304
+ [WARN] [1733882062.407196]: Plan exhausted
305
+ [WARN] [1733882062.447055]: Plan exhausted
306
+ [WARN] [1733882062.487156]: Plan exhausted
307
+ [WARN] [1733882062.527073]: Plan exhausted
308
+ [WARN] [1733882062.567137]: Plan exhausted
309
+ [WARN] [1733882062.607140]: Plan exhausted
310
+ [WARN] [1733882062.647053]: Plan exhausted
311
+ [WARN] [1733882062.687056]: Plan exhausted
312
+ [WARN] [1733882062.727053]: Plan exhausted
313
+ [WARN] [1733882062.767071]: Plan exhausted
314
+ [WARN] [1733882062.807082]: Plan exhausted
315
+ [WARN] [1733882062.847049]: Plan exhausted
316
+ [WARN] [1733882062.887068]: Plan exhausted
317
+ [WARN] [1733882062.927056]: Plan exhausted
318
+ [WARN] [1733882062.967074]: Plan exhausted
319
+ [WARN] [1733882063.007052]: Plan exhausted
320
+ [WARN] [1733882063.047042]: Plan exhausted
321
+ [WARN] [1733882063.087048]: Plan exhausted
322
+ [WARN] [1733882063.127049]: Plan exhausted
323
+ [WARN] [1733882063.167047]: Plan exhausted
324
+ [WARN] [1733882063.207056]: Plan exhausted
325
+ [WARN] [1733882063.247158]: Plan exhausted
326
+ [WARN] [1733882063.287093]: Plan exhausted
327
+ [WARN] [1733882063.327050]: Plan exhausted
328
+ [WARN] [1733882063.367152]: Plan exhausted
329
+ [WARN] [1733882063.407056]: Plan exhausted
330
+ [WARN] [1733882063.447058]: Plan exhausted
331
+ [WARN] [1733882063.487025]: Plan exhausted
332
+ [WARN] [1733882063.527024]: Plan exhausted
333
+ [WARN] [1733882063.567053]: Plan exhausted
334
+ [WARN] [1733882063.607054]: Plan exhausted
335
+ [WARN] [1733882063.647300]: Plan exhausted
336
+ [WARN] [1733882063.687048]: Plan exhausted
337
+ [WARN] [1733882063.727015]: Plan exhausted
338
+ [WARN] [1733882063.767130]: Plan exhausted
339
+ [WARN] [1733882063.807046]: Plan exhausted
340
+ [WARN] [1733882063.847275]: Plan exhausted
341
+ [WARN] [1733882063.887052]: Plan exhausted
342
+ [WARN] [1733882063.927107]: Plan exhausted
343
+ [WARN] [1733882063.967128]: Plan exhausted
344
+ [WARN] [1733882064.007052]: Plan exhausted
345
+ [WARN] [1733882064.047224]: Plan exhausted
346
+ [WARN] [1733882064.087026]: Plan exhausted
347
+ [WARN] [1733882064.127050]: Plan exhausted
348
+ [WARN] [1733882064.167024]: Plan exhausted
349
+ [WARN] [1733882064.207025]: Plan exhausted
350
+ [WARN] [1733882064.247044]: Plan exhausted
351
+ [WARN] [1733882064.287042]: Plan exhausted
352
+ [WARN] [1733882064.327054]: Plan exhausted
353
+ [WARN] [1733882064.367053]: Plan exhausted
354
+ [WARN] [1733882064.407223]: Plan exhausted
355
+ [WARN] [1733882064.447106]: Plan exhausted
356
+ [WARN] [1733882064.487052]: Plan exhausted
357
+ [WARN] [1733882064.527056]: Plan exhausted
358
+ [WARN] [1733882064.567079]: Plan exhausted
359
+ [WARN] [1733882064.607060]: Plan exhausted
360
+ [WARN] [1733882064.647050]: Plan exhausted
361
+ [WARN] [1733882064.687174]: Plan exhausted
362
+ [WARN] [1733882064.727053]: Plan exhausted
363
+ [WARN] [1733882064.767020]: Plan exhausted
364
+ [WARN] [1733882064.807027]: Plan exhausted
365
+ [WARN] [1733882064.847050]: Plan exhausted
366
+ [WARN] [1733882064.887045]: Plan exhausted
367
+ [WARN] [1733882064.927509]: Plan exhausted
368
+ [WARN] [1733882064.967138]: Plan exhausted
369
+ [WARN] [1733882065.007050]: Plan exhausted
370
+ [WARN] [1733882065.047022]: Plan exhausted
371
+ [WARN] [1733882065.087052]: Plan exhausted
372
+ [WARN] [1733882065.127049]: Plan exhausted
373
+ [WARN] [1733882065.167045]: Plan exhausted
374
+ [WARN] [1733882065.207147]: Plan exhausted
375
+ [WARN] [1733882065.247191]: Plan exhausted
376
+ [WARN] [1733882065.287047]: Plan exhausted
377
+ [WARN] [1733882065.327043]: Plan exhausted
378
+ [WARN] [1733882065.367355]: Plan exhausted
379
+ [WARN] [1733882065.407120]: Plan exhausted
380
+ [WARN] [1733882065.447181]: Plan exhausted
381
+ [WARN] [1733882065.487173]: Plan exhausted
382
+ [WARN] [1733882065.527201]: Plan exhausted
383
+ [WARN] [1733882065.567048]: Plan exhausted
384
+ [WARN] [1733882065.607166]: Plan exhausted
385
+ [WARN] [1733882065.647006]: Plan exhausted
386
+ [WARN] [1733882065.687023]: Plan exhausted
387
+ [WARN] [1733882065.727056]: Plan exhausted
388
+ [WARN] [1733882065.767043]: Plan exhausted
389
+ [WARN] [1733882065.807048]: Plan exhausted
390
+ [WARN] [1733882065.847017]: Plan exhausted
391
+ [WARN] [1733882065.887288]: Plan exhausted
392
+ [WARN] [1733882065.927184]: Plan exhausted
393
+ [WARN] [1733882065.967075]: Plan exhausted
394
+ [WARN] [1733882066.007171]: Plan exhausted
395
+ [WARN] [1733882066.047332]: Plan exhausted
396
+ [WARN] [1733882066.087051]: Plan exhausted
397
+ [WARN] [1733882066.127206]: Plan exhausted
398
+ [WARN] [1733882066.167326]: Plan exhausted
399
+ [WARN] [1733882066.207052]: Plan exhausted
400
+ [WARN] [1733882066.247040]: Plan exhausted
401
+ [WARN] [1733882066.287051]: Plan exhausted
402
+ [WARN] [1733882066.327026]: Plan exhausted
403
+ [WARN] [1733882066.367097]: Plan exhausted
404
+ [WARN] [1733882066.407017]: Plan exhausted
405
+ [WARN] [1733882066.447052]: Plan exhausted
406
+ [WARN] [1733882066.487055]: Plan exhausted
407
+ [WARN] [1733882066.527056]: Plan exhausted
408
+ [WARN] [1733882066.567098]: Plan exhausted
409
+ [WARN] [1733882066.607184]: Plan exhausted
410
+ [WARN] [1733882066.647047]: Plan exhausted
411
+ [WARN] [1733882066.687049]: Plan exhausted
412
+ [WARN] [1733882066.727019]: Plan exhausted
413
+ [WARN] [1733882066.767239]: Plan exhausted
414
+ [WARN] [1733882066.807016]: Plan exhausted
415
+ [WARN] [1733882066.847045]: Plan exhausted
416
+ [WARN] [1733882066.887052]: Plan exhausted
417
+ [WARN] [1733882066.927155]: Plan exhausted
418
+ [WARN] [1733882066.967067]: Plan exhausted
419
+ [WARN] [1733882067.007022]: Plan exhausted
420
+ [WARN] [1733882067.047061]: Plan exhausted
421
+ [WARN] [1733882067.087224]: Plan exhausted
422
+ [WARN] [1733882067.127199]: Plan exhausted
423
+ [WARN] [1733882067.167049]: Plan exhausted
424
+ [WARN] [1733882067.207051]: Plan exhausted
425
+ [WARN] [1733882067.247069]: Plan exhausted
426
+ [WARN] [1733882067.287317]: Plan exhausted
427
+ [WARN] [1733882067.327368]: Plan exhausted
428
+ [WARN] [1733882067.367091]: Plan exhausted
429
+ [WARN] [1733882067.407023]: Plan exhausted
430
+ [WARN] [1733882067.447050]: Plan exhausted
431
+ [WARN] [1733882067.487056]: Plan exhausted
432
+ [WARN] [1733882067.527055]: Plan exhausted
433
+ [WARN] [1733882067.578561]: Plan exhausted
434
+ [WARN] [1733882067.607018]: Plan exhausted
435
+ [WARN] [1733882067.647185]: Plan exhausted
436
+ Exception in thread Thread-62:
437
+ Traceback (most recent call last):
438
+ File "/home/leonmkim/.pyenv/versions/3.10.14/lib/python3.10/threading.py", line 1016, in _bootstrap_inner
439
+ self.run()
440
+ File "/opt/ros/noetic/lib/python3/dist-packages/rospy/timer.py", line 237, in run
441
+ self._callback(TimerEvent(last_expected, last_real, current_expected, current_real, last_duration))
442
+ File "/home/leonmkim/fish_leon/FISH/video.py", line 78, in ros_record_cb
443
+ self.process.stdin.write(frame.tobytes())
444
+ BrokenPipeError: [Errno 32] Broken pipe
445
+ [WARN] [1733882067.687189]: Plan exhausted
446
+ [WARN] [1733882067.727179]: Plan exhausted
447
+ [WARN] [1733882067.767038]: Plan exhausted
448
+ [WARN] [1733882067.807042]: Plan exhausted
449
+ [WARN] [1733882067.847068]: Plan exhausted
450
+ [WARN] [1733882067.887222]: Plan exhausted
451
+ [WARN] [1733882067.927459]: Plan exhausted
452
+ [WARN] [1733882067.967060]: Plan exhausted
453
+ [WARN] [1733882068.007040]: Plan exhausted
454
+ [WARN] [1733882068.047081]: Plan exhausted
455
+ [rosbag_record-3] killing on exit
456
+ [print_text-2] killing on exit
457
+ [pub_text-1] killing on exit
458
+ program exiting gracefully
459
+ proceeding to start episode!
460
+ [INFO] [1733882069.242753]: Saving video...
461
+ [INFO] [1733882069.243578]: Video saved!
205324/wandb/run-20241210_205330-52nddla7/files/requirements.txt ADDED
@@ -0,0 +1,339 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ Cython==3.0.10
2
+ Farama-Notifications==0.0.4
3
+ GitPython==3.1.43
4
+ Jinja2==3.1.4
5
+ Markdown==3.6
6
+ MarkupSafe==2.1.5
7
+ POT==0.7.0
8
+ PyOpenGL==3.1.7
9
+ PySocks==1.7.1
10
+ PyYAML==6.0.1
11
+ Pygments==2.18.0
12
+ Rtree==1.3.0
13
+ Werkzeug==3.0.3
14
+ absl-py==2.1.0
15
+ accelerate==0.33.0
16
+ actionlib-msgs==1.13.0.post3
17
+ actionlib==1.12.0
18
+ actionlib==1.14.0
19
+ aiohttp==3.9.5
20
+ aiosignal==1.3.1
21
+ angles==1.9.13
22
+ antlr4-python3-runtime==4.9.3
23
+ anyio==4.4.0
24
+ asciitree==0.3.3
25
+ async-timeout==4.0.3
26
+ attrs==23.2.0
27
+ autoprop==4.1.0
28
+ beartype==0.18.5
29
+ beautifulsoup4==4.12.3
30
+ bondpy==1.8.6
31
+ byol-pytorch==0.8.0
32
+ cachetools==5.4.0
33
+ camera-calibration-parsers==1.12.0
34
+ camera-calibration==1.17.0
35
+ cascadio==0.0.13
36
+ catkin-pkg==1.0.0
37
+ catkin==0.7.18
38
+ catkin==0.8.10
39
+ certifi==2024.7.4
40
+ cffi==1.16.0
41
+ chardet==5.2.0
42
+ charset-normalizer==3.3.2
43
+ click==8.1.7
44
+ cloudpickle==3.0.0
45
+ cmake==3.30.1
46
+ colorlog==6.8.2
47
+ contourpy==1.2.1
48
+ controller-manager-msgs==0.20.0
49
+ controller-manager==0.20.0
50
+ coverage==7.6.0
51
+ coveralls==4.0.1
52
+ cv-bridge==1.16.2
53
+ cycler==0.12.1
54
+ datasets==2.20.0
55
+ decorator==4.4.2
56
+ deepdiff==7.0.1
57
+ defusedxml==0.7.1
58
+ diagnostic-analysis==1.11.0
59
+ diagnostic-common-diagnostics==1.11.0
60
+ diagnostic-updater==1.11.0
61
+ diffusers==0.27.2
62
+ dill==0.3.8
63
+ distro==1.9.0
64
+ dm-control==1.0.8
65
+ dm-env==1.6
66
+ dm-tree==0.1.8
67
+ docker-pycreds==0.4.0
68
+ docopt==0.6.2
69
+ docutils==0.21.2
70
+ dynamic-reconfigure==1.7.3
71
+ einops==0.8.0
72
+ embreex==2.17.7.post5
73
+ empy==3.3.4
74
+ etils==1.7.0
75
+ exceptiongroup==1.2.2
76
+ ezdxf==1.3.2
77
+ fasteners==0.19
78
+ filelock==3.15.4
79
+ fonttools==4.53.1
80
+ freetype-py==2.4.0
81
+ frozenlist==1.4.1
82
+ fsspec==2024.5.0
83
+ gazebo_plugins==2.9.2
84
+ gazebo_ros==2.9.2
85
+ gdown==5.2.0
86
+ gencpp==0.7.0
87
+ geneus==3.0.0
88
+ genlisp==0.4.18
89
+ genmsg==0.5.12
90
+ genmsg==0.6.0
91
+ gennodejs==2.0.2
92
+ genpy==0.6.14
93
+ genpy==0.6.15
94
+ geometry-msgs==1.13.0.post2
95
+ gitdb==4.0.11
96
+ glfw==2.7.0
97
+ glooey==0.3.6
98
+ gmsh==4.12.2
99
+ gnupg==2.3.1
100
+ google-auth-oauthlib==1.0.0
101
+ google-auth==2.32.0
102
+ grpcio==1.65.1
103
+ gym-envs==0.0.1
104
+ gym-notices==0.0.8
105
+ gym==0.22.0
106
+ gymnasium==0.29.1
107
+ h11==0.14.0
108
+ h5py==3.11.0
109
+ hf_transfer==0.1.8
110
+ httpcore==1.0.5
111
+ httpx==0.27.0
112
+ huggingface-hub==0.23.5
113
+ hydra-core==1.3.2
114
+ hydra-submitit-launcher==1.2.0
115
+ idna==3.7
116
+ image-geometry==1.16.2
117
+ imageio-ffmpeg==0.5.1
118
+ imageio==2.34.2
119
+ importlib_metadata==8.2.0
120
+ importlib_resources==6.4.0
121
+ iniconfig==2.0.0
122
+ interactive-markers==1.12.0
123
+ joint-state-publisher-gui==1.15.1
124
+ joint-state-publisher==1.15.1
125
+ jsonschema-specifications==2023.12.1
126
+ jsonschema==4.23.0
127
+ kiwisolver==1.4.5
128
+ kornia==0.7.3
129
+ kornia_rs==0.1.5
130
+ labmaze==1.0.6
131
+ laser_geometry==1.6.7
132
+ lazy_loader==0.4
133
+ lerobot==0.1.0
134
+ lightning-utilities==0.11.6
135
+ llvmlite==0.43.0
136
+ lxml==5.2.2
137
+ manifold3d==2.5.1
138
+ mapbox-earcut==1.0.1
139
+ markdown-it-py==3.0.0
140
+ matplotlib==3.9.1
141
+ mdurl==0.1.2
142
+ meshio==5.3.5
143
+ message-filters==1.16.0
144
+ more-itertools==10.3.0
145
+ moviepy==1.0.3
146
+ mpmath==1.3.0
147
+ mujoco==3.2.0
148
+ multidict==6.0.5
149
+ multiprocess==0.70.16
150
+ natsort==8.4.0
151
+ netifaces==0.11.0
152
+ networkx==3.3
153
+ nodeenv==1.9.1
154
+ numba==0.60.0
155
+ numcodecs==0.13.0
156
+ numpy==1.26.4
157
+ nvidia-cublas-cu12==12.1.3.1
158
+ nvidia-cuda-cupti-cu12==12.1.105
159
+ nvidia-cuda-nvrtc-cu12==12.1.105
160
+ nvidia-cuda-runtime-cu12==12.1.105
161
+ nvidia-cudnn-cu12==9.1.0.70
162
+ nvidia-cufft-cu12==11.0.2.54
163
+ nvidia-curand-cu12==10.3.2.106
164
+ nvidia-cusolver-cu12==11.4.5.107
165
+ nvidia-cusparse-cu12==12.1.0.106
166
+ nvidia-nccl-cu12==2.20.5
167
+ nvidia-nvjitlink-cu12==12.5.82
168
+ nvidia-nvtx-cu12==12.1.105
169
+ oauthlib==3.2.2
170
+ omegaconf==2.3.0
171
+ openctm==0.0.5
172
+ opencv-python==4.10.0.84
173
+ ordered-set==4.1.0
174
+ packaging==24.1
175
+ pandas==2.2.2
176
+ pillow==10.4.0
177
+ pip==24.2
178
+ platformdirs==4.2.2
179
+ pluggy==1.5.0
180
+ proglog==0.1.10
181
+ protobuf==5.27.2
182
+ psutil==6.0.0
183
+ pyarrow-hotfix==0.6
184
+ pyarrow==17.0.0
185
+ pyasn1==0.6.0
186
+ pyasn1_modules==0.4.0
187
+ pyav==12.3.0
188
+ pycollada==0.8
189
+ pycparser==2.22
190
+ pycryptodomex==3.21.0
191
+ pyglet==1.5.29
192
+ pyinstrument==4.6.2
193
+ pymunk==6.8.1
194
+ pyparsing==2.4.7
195
+ pyrealsense2==2.54.2.5684
196
+ pyribbit==0.1.46
197
+ pyright==1.1.373
198
+ pytest-beartype==0.0.2
199
+ pytest-cov==5.0.0
200
+ pytest==8.3.1
201
+ python-dateutil==2.9.0.post0
202
+ python-fcl==0.7.0.6
203
+ python-qt-binding==0.4.4
204
+ pytorch-lightning==2.4.0
205
+ pytz==2024.1
206
+ qt-dotgraph==0.4.2
207
+ qt-gui-cpp==0.4.2
208
+ qt-gui-py-common==0.4.2
209
+ qt-gui==0.4.2
210
+ referencing==0.35.1
211
+ regex==2024.5.15
212
+ requests-oauthlib==2.0.0
213
+ requests==2.32.3
214
+ rerun-sdk==0.17.0
215
+ resource_retriever==1.12.7
216
+ rich==13.7.1
217
+ ros-numpy==0.0.5
218
+ rosbag==1.16.0
219
+ rosboost-cfg==1.15.8
220
+ rosclean==1.15.8
221
+ roscpp==1.15.11
222
+ roscreate==1.15.8
223
+ rosgraph-msgs==1.11.3.post2
224
+ rosgraph==1.15.11
225
+ rosgraph==1.16.0
226
+ roslaunch==1.16.0
227
+ roslib==1.14.7.post0
228
+ roslib==1.15.8
229
+ roslint==0.12.0
230
+ roslz4==1.16.0
231
+ rosmake==1.15.8
232
+ rosmaster==1.16.0
233
+ rosmsg==1.16.0
234
+ rosnode==1.16.0
235
+ rosparam==1.16.0
236
+ rospkg==1.5.1
237
+ rospy==1.15.11
238
+ rospy==1.16.0
239
+ rosservice==1.16.0
240
+ rostest==1.16.0
241
+ rostopic==1.16.0
242
+ rosunit==1.15.8
243
+ roswtf==1.16.0
244
+ rpds-py==0.19.1
245
+ rqt-console==0.4.12
246
+ rqt-image-view==0.4.17
247
+ rqt-logger-level==0.4.12
248
+ rqt-moveit==0.5.11
249
+ rqt-reconfigure==0.5.5
250
+ rqt-robot-dashboard==0.5.8
251
+ rqt-robot-monitor==0.5.15
252
+ rqt-runtime-monitor==0.5.10
253
+ rqt-rviz==0.7.0
254
+ rqt-tf-tree==0.6.4
255
+ rqt_action==0.4.9
256
+ rqt_bag==0.5.1
257
+ rqt_bag_plugins==0.5.1
258
+ rqt_dep==0.4.12
259
+ rqt_graph==0.4.14
260
+ rqt_gui==0.5.3
261
+ rqt_gui_py==0.5.3
262
+ rqt_launch==0.4.9
263
+ rqt_msg==0.4.10
264
+ rqt_nav_view==0.5.7
265
+ rqt_plot==0.4.13
266
+ rqt_pose_view==0.5.11
267
+ rqt_publisher==0.4.10
268
+ rqt_py_common==0.5.3
269
+ rqt_py_console==0.4.10
270
+ rqt_robot_steering==0.5.12
271
+ rqt_service_caller==0.4.10
272
+ rqt_shell==0.4.11
273
+ rqt_srv==0.4.9
274
+ rqt_top==0.4.10
275
+ rqt_topic==0.4.13
276
+ rqt_web==0.4.10
277
+ rsa==4.9
278
+ ruff==0.5.4
279
+ rviz==1.14.25
280
+ safetensors==0.4.3
281
+ scikit-image==0.24.0
282
+ scikit-video==1.1.11
283
+ scipy==1.14.0
284
+ seaborn==0.13.2
285
+ sensor-msgs==1.13.1
286
+ sentry-sdk==2.11.0
287
+ setproctitle==1.3.3
288
+ setuptools==65.5.0
289
+ shapely==2.0.5
290
+ signature_dispatch==1.0.1
291
+ six==1.16.0
292
+ smach-ros==2.5.2
293
+ smach==2.5.2
294
+ smclib==1.8.6
295
+ smmap==5.0.1
296
+ sniffio==1.3.1
297
+ soupsieve==2.5
298
+ std-msgs==0.5.13.post0
299
+ submitit==1.5.1
300
+ svg.path==6.3
301
+ sympy==1.13.1
302
+ tensorboard-data-server==0.7.2
303
+ tensorboard==2.14.0
304
+ termcolor==2.4.0
305
+ tf-conversions==1.13.2
306
+ tf2-geometry-msgs==0.7.7
307
+ tf2-kdl==0.7.7
308
+ tf2-msgs==0.7.2.post3
309
+ tf2-py==0.7.7
310
+ tf2-ros==0.6.5
311
+ tf2-ros==0.7.7
312
+ tf2_py==0.6.5.post1
313
+ tf==1.13.2
314
+ tifffile==2024.7.24
315
+ tomli==2.0.1
316
+ topic-tools==1.16.0
317
+ torch==2.4.0
318
+ torchaug==0.5.2
319
+ torchmetrics==1.4.0.post0
320
+ torchvision==0.19.0
321
+ tqdm==4.66.4
322
+ trimesh==4.4.3
323
+ triton==3.0.0
324
+ typeguard==3.0.2
325
+ typing_extensions==4.12.2
326
+ tzdata==2024.1
327
+ urchin==0.0.27
328
+ urllib3==2.2.2
329
+ vecrec==0.3.1
330
+ vhacdx==0.0.8.post1
331
+ wandb==0.17.5
332
+ wheel==0.43.0
333
+ xacro==1.14.18
334
+ xatlas==0.0.9
335
+ xxhash==3.4.1
336
+ yarl==1.9.4
337
+ zarr==2.18.2
338
+ zipp==3.19.2
339
+ zstandard==0.23.0
205324/wandb/run-20241210_205330-52nddla7/files/wandb-metadata.json ADDED
@@ -0,0 +1,92 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "os": "Linux-5.15.0-125-generic-x86_64-with-glibc2.31",
3
+ "python": "3.10.14",
4
+ "heartbeatAt": "2024-12-11T01:53:31.224621",
5
+ "startedAt": "2024-12-11T01:53:30.842354",
6
+ "docker": null,
7
+ "cuda": null,
8
+ "args": [
9
+ "agent=diffusion",
10
+ "suite=frankagym",
11
+ "suite/frankagym_task@_global_=insertion"
12
+ ],
13
+ "state": "running",
14
+ "program": "/home/leonmkim/fish_leon/FISH/eval_robot.py",
15
+ "codePathLocal": null,
16
+ "codePath": "FISH/eval_robot.py",
17
+ "git": {
18
+ "remote": "https://github.com/leonmkim/fish_leon.git",
19
+ "commit": "4f7c0421eb84a7caf6bf5670f68777523362676f"
20
+ },
21
+ "email": "leonmkim@seas.upenn.edu",
22
+ "root": "/home/leonmkim/fish_leon",
23
+ "host": "leonmkim-ROG-Strix-G15CS-G15CS",
24
+ "username": "leonmkim",
25
+ "executable": "/home/leonmkim/.pyenv/versions/lerobot/bin/python",
26
+ "cpu_count": 8,
27
+ "cpu_count_logical": 8,
28
+ "cpu_freq": {
29
+ "current": 4312.5025,
30
+ "min": 800.0,
31
+ "max": 4700.0
32
+ },
33
+ "cpu_freq_per_core": [
34
+ {
35
+ "current": 4500.019,
36
+ "min": 800.0,
37
+ "max": 4700.0
38
+ },
39
+ {
40
+ "current": 4500.002,
41
+ "min": 800.0,
42
+ "max": 4700.0
43
+ },
44
+ {
45
+ "current": 3000.0,
46
+ "min": 800.0,
47
+ "max": 4700.0
48
+ },
49
+ {
50
+ "current": 4500.007,
51
+ "min": 800.0,
52
+ "max": 4700.0
53
+ },
54
+ {
55
+ "current": 4499.999,
56
+ "min": 800.0,
57
+ "max": 4700.0
58
+ },
59
+ {
60
+ "current": 3000.0,
61
+ "min": 800.0,
62
+ "max": 4700.0
63
+ },
64
+ {
65
+ "current": 3000.0,
66
+ "min": 800.0,
67
+ "max": 4700.0
68
+ },
69
+ {
70
+ "current": 4499.943,
71
+ "min": 800.0,
72
+ "max": 4700.0
73
+ }
74
+ ],
75
+ "disk": {
76
+ "/": {
77
+ "total": 915.3232879638672,
78
+ "used": 365.2335090637207
79
+ }
80
+ },
81
+ "gpu": "NVIDIA GeForce RTX 2070 SUPER",
82
+ "gpu_count": 1,
83
+ "gpu_devices": [
84
+ {
85
+ "name": "NVIDIA GeForce RTX 2070 SUPER",
86
+ "memory_total": 8589934592
87
+ }
88
+ ],
89
+ "memory": {
90
+ "total": 62.71596908569336
91
+ }
92
+ }
205324/wandb/run-20241210_205330-52nddla7/files/wandb-summary.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"eval/video": {"_type": "video-file", "sha256": "ff4edecb6a17ea039fbf12ba8ab1396b6c7798607c09ee107432df34eedac9aa", "size": 622320, "path": "media/videos/eval/video_0_ff4edecb6a17ea039fbf.mp4"}, "global_step": 1, "_timestamp": 1733882069.2541587, "_runtime": 58.40333962440491, "_step": 0, "_wandb": {"runtime": 58}}
205324/wandb/run-20241210_205330-52nddla7/logs/debug-internal.log ADDED
@@ -0,0 +1,327 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ 2024-12-10 20:53:30,850 INFO StreamThr :598006 [internal.py:wandb_internal():85] W&B internal server running at pid: 598006, started at: 2024-12-10 20:53:30.850039
2
+ 2024-12-10 20:53:30,851 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: status
3
+ 2024-12-10 20:53:30,863 INFO WriterThread:598006 [datastore.py:open_for_write():87] open: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205324/wandb/run-20241210_205330-52nddla7/run-52nddla7.wandb
4
+ 2024-12-10 20:53:30,863 DEBUG SenderThread:598006 [sender.py:send():379] send: header
5
+ 2024-12-10 20:53:30,875 DEBUG SenderThread:598006 [sender.py:send():379] send: run
6
+ 2024-12-10 20:53:31,147 INFO SenderThread:598006 [dir_watcher.py:__init__():211] watching files in: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205324/wandb/run-20241210_205330-52nddla7/files
7
+ 2024-12-10 20:53:31,147 INFO SenderThread:598006 [sender.py:_start_run_threads():1188] run started: 52nddla7 with start time 1733882010.850819
8
+ 2024-12-10 20:53:31,161 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: check_version
9
+ 2024-12-10 20:53:31,161 DEBUG SenderThread:598006 [sender.py:send_request():406] send_request: check_version
10
+ 2024-12-10 20:53:31,206 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: run_start
11
+ 2024-12-10 20:53:31,211 DEBUG HandlerThread:598006 [system_info.py:__init__():26] System info init
12
+ 2024-12-10 20:53:31,211 DEBUG HandlerThread:598006 [system_info.py:__init__():41] System info init done
13
+ 2024-12-10 20:53:31,211 INFO HandlerThread:598006 [system_monitor.py:start():194] Starting system monitor
14
+ 2024-12-10 20:53:31,211 INFO SystemMonitor:598006 [system_monitor.py:_start():158] Starting system asset monitoring threads
15
+ 2024-12-10 20:53:31,212 INFO HandlerThread:598006 [system_monitor.py:probe():214] Collecting system info
16
+ 2024-12-10 20:53:31,212 INFO SystemMonitor:598006 [interfaces.py:start():188] Started cpu monitoring
17
+ 2024-12-10 20:53:31,212 INFO SystemMonitor:598006 [interfaces.py:start():188] Started disk monitoring
18
+ 2024-12-10 20:53:31,213 INFO SystemMonitor:598006 [interfaces.py:start():188] Started gpu monitoring
19
+ 2024-12-10 20:53:31,213 INFO SystemMonitor:598006 [interfaces.py:start():188] Started memory monitoring
20
+ 2024-12-10 20:53:31,214 INFO SystemMonitor:598006 [interfaces.py:start():188] Started network monitoring
21
+ 2024-12-10 20:53:31,224 DEBUG HandlerThread:598006 [system_info.py:probe():152] Probing system
22
+ 2024-12-10 20:53:31,226 DEBUG HandlerThread:598006 [system_info.py:_probe_git():137] Probing git
23
+ 2024-12-10 20:53:31,231 DEBUG HandlerThread:598006 [system_info.py:_probe_git():145] Probing git done
24
+ 2024-12-10 20:53:31,231 DEBUG HandlerThread:598006 [system_info.py:probe():200] Probing system done
25
+ 2024-12-10 20:53:31,231 DEBUG HandlerThread:598006 [system_monitor.py:probe():223] {'os': 'Linux-5.15.0-125-generic-x86_64-with-glibc2.31', 'python': '3.10.14', 'heartbeatAt': '2024-12-11T01:53:31.224621', 'startedAt': '2024-12-11T01:53:30.842354', 'docker': None, 'cuda': None, 'args': ('agent=diffusion', 'suite=frankagym', 'suite/frankagym_task@_global_=insertion'), 'state': 'running', 'program': '/home/leonmkim/fish_leon/FISH/eval_robot.py', 'codePathLocal': None, 'codePath': 'FISH/eval_robot.py', 'git': {'remote': 'https://github.com/leonmkim/fish_leon.git', 'commit': '4f7c0421eb84a7caf6bf5670f68777523362676f'}, 'email': 'leonmkim@seas.upenn.edu', 'root': '/home/leonmkim/fish_leon', 'host': 'leonmkim-ROG-Strix-G15CS-G15CS', 'username': 'leonmkim', 'executable': '/home/leonmkim/.pyenv/versions/lerobot/bin/python', 'cpu_count': 8, 'cpu_count_logical': 8, 'cpu_freq': {'current': 4312.5025, 'min': 800.0, 'max': 4700.0}, 'cpu_freq_per_core': [{'current': 4500.019, 'min': 800.0, 'max': 4700.0}, {'current': 4500.002, 'min': 800.0, 'max': 4700.0}, {'current': 3000.0, 'min': 800.0, 'max': 4700.0}, {'current': 4500.007, 'min': 800.0, 'max': 4700.0}, {'current': 4499.999, 'min': 800.0, 'max': 4700.0}, {'current': 3000.0, 'min': 800.0, 'max': 4700.0}, {'current': 3000.0, 'min': 800.0, 'max': 4700.0}, {'current': 4499.943, 'min': 800.0, 'max': 4700.0}], 'disk': {'/': {'total': 915.3232879638672, 'used': 365.2335090637207}}, 'gpu': 'NVIDIA GeForce RTX 2070 SUPER', 'gpu_count': 1, 'gpu_devices': [{'name': 'NVIDIA GeForce RTX 2070 SUPER', 'memory_total': 8589934592}], 'memory': {'total': 62.71596908569336}}
26
+ 2024-12-10 20:53:31,231 INFO HandlerThread:598006 [system_monitor.py:probe():224] Finished collecting system info
27
+ 2024-12-10 20:53:31,231 INFO HandlerThread:598006 [system_monitor.py:probe():227] Publishing system info
28
+ 2024-12-10 20:53:31,231 DEBUG HandlerThread:598006 [system_info.py:_save_code():44] Saving code
29
+ 2024-12-10 20:53:31,233 DEBUG HandlerThread:598006 [system_info.py:_save_code():67] Saving code done
30
+ 2024-12-10 20:53:31,233 DEBUG HandlerThread:598006 [system_info.py:_save_patches():84] Saving git patches
31
+ 2024-12-10 20:53:31,305 DEBUG HandlerThread:598006 [system_info.py:_save_patches():126] Saving git patches done
32
+ 2024-12-10 20:53:31,306 INFO HandlerThread:598006 [system_monitor.py:probe():229] Finished publishing system info
33
+ 2024-12-10 20:53:31,309 DEBUG SenderThread:598006 [sender.py:send():379] send: files
34
+ 2024-12-10 20:53:31,310 INFO SenderThread:598006 [sender.py:_save_file():1454] saving file wandb-metadata.json with policy now
35
+ 2024-12-10 20:53:31,310 INFO SenderThread:598006 [sender.py:_save_file():1454] saving file code/FISH/eval_robot.py with policy now
36
+ 2024-12-10 20:53:31,310 INFO SenderThread:598006 [sender.py:_save_file():1454] saving file diff.patch with policy now
37
+ 2024-12-10 20:53:31,570 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: python_packages
38
+ 2024-12-10 20:53:31,570 DEBUG SenderThread:598006 [sender.py:send_request():406] send_request: python_packages
39
+ 2024-12-10 20:53:31,570 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: stop_status
40
+ 2024-12-10 20:53:31,571 INFO wandb-upload_0:598006 [upload_job.py:push():130] Uploaded file /tmp/tmptbytljn2wandb/oqlixunj-wandb-metadata.json
41
+ 2024-12-10 20:53:31,571 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
42
+ 2024-12-10 20:53:31,572 DEBUG SenderThread:598006 [sender.py:send_request():406] send_request: stop_status
43
+ 2024-12-10 20:53:31,573 INFO HandlerThread:598006 [handler.py:handle_tbrecord():783] handling tbrecord: tbrecord {
44
+ log_dir: "/home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205324/tb"
45
+ save: true
46
+ }
47
+ control {
48
+ relay_id: "5d813b8bb5a547c8896a7e03615dca84"
49
+ }
50
+ _info {
51
+ stream_id: "52nddla7"
52
+ }
53
+
54
+ 2024-12-10 20:53:31,574 DEBUG HandlerThread:598006 [config_util.py:dict_from_config_file():70] no default config file found in config-defaults.yaml
55
+ 2024-12-10 20:53:31,605 INFO wandb-upload_1:598006 [upload_job.py:push():130] Uploaded file /tmp/tmptbytljn2wandb/6yv6eek3-code/FISH/eval_robot.py
56
+ 2024-12-10 20:53:31,648 INFO wandb-upload_2:598006 [upload_job.py:push():130] Uploaded file /tmp/tmptbytljn2wandb/w33l6u66-diff.patch
57
+ 2024-12-10 20:53:31,702 DEBUG SenderThread:598006 [sender.py:send():379] send: telemetry
58
+ 2024-12-10 20:53:31,702 DEBUG SenderThread:598006 [sender.py:send():379] send: tbrecord
59
+ 2024-12-10 20:53:31,703 DEBUG SenderThread:598006 [sender.py:send():379] send: files
60
+ 2024-12-10 20:53:31,703 INFO SenderThread:598006 [sender.py:_save_file():1454] saving file events.out.tfevents.1733882011.leonmkim-ROG-Strix-G15CS-G15CS.597884.0 with policy live
61
+ 2024-12-10 20:53:31,907 INFO wandb-upload_0:598006 [upload_job.py:push():130] Uploaded file /tmp/tmptbytljn2wandb/uat8jiac-events.out.tfevents.1733882011.leonmkim-ROG-Strix-G15CS-G15CS.597884.0
62
+ 2024-12-10 20:53:32,149 INFO Thread-12 :598006 [dir_watcher.py:_on_file_created():271] file/dir created: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205324/wandb/run-20241210_205330-52nddla7/files/events.out.tfevents.1733882011.leonmkim-ROG-Strix-G15CS-G15CS.597884.0
63
+ 2024-12-10 20:53:32,149 INFO Thread-12 :598006 [dir_watcher.py:_on_file_created():271] file/dir created: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205324/wandb/run-20241210_205330-52nddla7/files/code/FISH/eval_robot.py
64
+ 2024-12-10 20:53:32,149 INFO Thread-12 :598006 [dir_watcher.py:_on_file_created():271] file/dir created: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205324/wandb/run-20241210_205330-52nddla7/files/wandb-metadata.json
65
+ 2024-12-10 20:53:32,149 INFO Thread-12 :598006 [dir_watcher.py:_on_file_created():271] file/dir created: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205324/wandb/run-20241210_205330-52nddla7/files/output.log
66
+ 2024-12-10 20:53:32,149 INFO Thread-12 :598006 [dir_watcher.py:_on_file_created():271] file/dir created: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205324/wandb/run-20241210_205330-52nddla7/files/requirements.txt
67
+ 2024-12-10 20:53:32,149 INFO Thread-12 :598006 [dir_watcher.py:_on_file_created():271] file/dir created: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205324/wandb/run-20241210_205330-52nddla7/files/diff.patch
68
+ 2024-12-10 20:53:32,149 INFO Thread-12 :598006 [dir_watcher.py:_on_file_created():271] file/dir created: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205324/wandb/run-20241210_205330-52nddla7/files/code/FISH
69
+ 2024-12-10 20:53:32,149 INFO Thread-12 :598006 [dir_watcher.py:_on_file_created():271] file/dir created: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205324/wandb/run-20241210_205330-52nddla7/files/code
70
+ 2024-12-10 20:53:32,571 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
71
+ 2024-12-10 20:53:33,571 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
72
+ 2024-12-10 20:53:34,148 INFO Thread-12 :598006 [dir_watcher.py:_on_file_modified():288] file/dir modified: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205324/wandb/run-20241210_205330-52nddla7/files/output.log
73
+ 2024-12-10 20:53:34,571 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
74
+ 2024-12-10 20:53:35,547 DEBUG SenderThread:598006 [sender.py:send():379] send: config
75
+ 2024-12-10 20:53:35,590 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
76
+ 2024-12-10 20:53:36,571 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: status_report
77
+ 2024-12-10 20:53:36,572 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
78
+ 2024-12-10 20:53:37,572 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
79
+ 2024-12-10 20:53:38,149 INFO Thread-12 :598006 [dir_watcher.py:_on_file_modified():288] file/dir modified: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205324/wandb/run-20241210_205330-52nddla7/files/output.log
80
+ 2024-12-10 20:53:38,572 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
81
+ 2024-12-10 20:53:39,572 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
82
+ 2024-12-10 20:53:40,150 INFO Thread-12 :598006 [dir_watcher.py:_on_file_modified():288] file/dir modified: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205324/wandb/run-20241210_205330-52nddla7/files/output.log
83
+ 2024-12-10 20:53:40,572 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
84
+ 2024-12-10 20:53:41,572 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
85
+ 2024-12-10 20:53:42,082 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: status_report
86
+ 2024-12-10 20:53:42,150 INFO Thread-12 :598006 [dir_watcher.py:_on_file_modified():288] file/dir modified: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205324/wandb/run-20241210_205330-52nddla7/files/output.log
87
+ 2024-12-10 20:53:42,572 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
88
+ 2024-12-10 20:53:43,572 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
89
+ 2024-12-10 20:53:44,151 INFO Thread-12 :598006 [dir_watcher.py:_on_file_modified():288] file/dir modified: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205324/wandb/run-20241210_205330-52nddla7/files/output.log
90
+ 2024-12-10 20:53:44,573 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
91
+ 2024-12-10 20:53:45,573 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
92
+ 2024-12-10 20:53:46,151 INFO Thread-12 :598006 [dir_watcher.py:_on_file_modified():288] file/dir modified: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205324/wandb/run-20241210_205330-52nddla7/files/output.log
93
+ 2024-12-10 20:53:46,570 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: stop_status
94
+ 2024-12-10 20:53:46,570 DEBUG SenderThread:598006 [sender.py:send_request():406] send_request: stop_status
95
+ 2024-12-10 20:53:46,614 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
96
+ 2024-12-10 20:53:47,573 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
97
+ 2024-12-10 20:53:47,640 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: status_report
98
+ 2024-12-10 20:53:48,573 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
99
+ 2024-12-10 20:53:49,573 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
100
+ 2024-12-10 20:53:50,573 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
101
+ 2024-12-10 20:53:51,574 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
102
+ 2024-12-10 20:53:52,574 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
103
+ 2024-12-10 20:53:53,571 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: status_report
104
+ 2024-12-10 20:53:53,574 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
105
+ 2024-12-10 20:53:54,574 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
106
+ 2024-12-10 20:53:55,574 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
107
+ 2024-12-10 20:53:56,574 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
108
+ 2024-12-10 20:53:57,574 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
109
+ 2024-12-10 20:53:58,571 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: status_report
110
+ 2024-12-10 20:53:58,574 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
111
+ 2024-12-10 20:53:59,575 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
112
+ 2024-12-10 20:54:00,575 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
113
+ 2024-12-10 20:54:01,570 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: stop_status
114
+ 2024-12-10 20:54:01,570 DEBUG SenderThread:598006 [sender.py:send_request():406] send_request: stop_status
115
+ 2024-12-10 20:54:01,613 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
116
+ 2024-12-10 20:54:02,575 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
117
+ 2024-12-10 20:54:03,575 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
118
+ 2024-12-10 20:54:03,722 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: status_report
119
+ 2024-12-10 20:54:04,156 INFO Thread-12 :598006 [dir_watcher.py:_on_file_modified():288] file/dir modified: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205324/wandb/run-20241210_205330-52nddla7/files/config.yaml
120
+ 2024-12-10 20:54:04,575 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
121
+ 2024-12-10 20:54:05,575 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
122
+ 2024-12-10 20:54:06,575 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
123
+ 2024-12-10 20:54:07,576 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
124
+ 2024-12-10 20:54:08,576 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
125
+ 2024-12-10 20:54:09,550 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: status_report
126
+ 2024-12-10 20:54:09,576 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
127
+ 2024-12-10 20:54:10,158 INFO Thread-12 :598006 [dir_watcher.py:_on_file_modified():288] file/dir modified: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205324/wandb/run-20241210_205330-52nddla7/files/output.log
128
+ 2024-12-10 20:54:10,576 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
129
+ 2024-12-10 20:54:11,576 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
130
+ 2024-12-10 20:54:12,159 INFO Thread-12 :598006 [dir_watcher.py:_on_file_modified():288] file/dir modified: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205324/wandb/run-20241210_205330-52nddla7/files/output.log
131
+ 2024-12-10 20:54:12,577 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
132
+ 2024-12-10 20:54:13,577 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
133
+ 2024-12-10 20:54:14,159 INFO Thread-12 :598006 [dir_watcher.py:_on_file_modified():288] file/dir modified: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205324/wandb/run-20241210_205330-52nddla7/files/output.log
134
+ 2024-12-10 20:54:14,567 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: status_report
135
+ 2024-12-10 20:54:14,577 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
136
+ 2024-12-10 20:54:15,160 INFO Thread-12 :598006 [dir_watcher.py:_on_file_modified():288] file/dir modified: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205324/wandb/run-20241210_205330-52nddla7/files/output.log
137
+ 2024-12-10 20:54:15,577 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
138
+ 2024-12-10 20:54:16,570 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: stop_status
139
+ 2024-12-10 20:54:16,570 DEBUG SenderThread:598006 [sender.py:send_request():406] send_request: stop_status
140
+ 2024-12-10 20:54:16,577 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
141
+ 2024-12-10 20:54:17,160 INFO Thread-12 :598006 [dir_watcher.py:_on_file_modified():288] file/dir modified: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205324/wandb/run-20241210_205330-52nddla7/files/output.log
142
+ 2024-12-10 20:54:17,577 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
143
+ 2024-12-10 20:54:18,579 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
144
+ 2024-12-10 20:54:19,161 INFO Thread-12 :598006 [dir_watcher.py:_on_file_modified():288] file/dir modified: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205324/wandb/run-20241210_205330-52nddla7/files/output.log
145
+ 2024-12-10 20:54:19,567 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: status_report
146
+ 2024-12-10 20:54:19,577 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
147
+ 2024-12-10 20:54:20,577 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
148
+ 2024-12-10 20:54:21,161 INFO Thread-12 :598006 [dir_watcher.py:_on_file_modified():288] file/dir modified: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205324/wandb/run-20241210_205330-52nddla7/files/output.log
149
+ 2024-12-10 20:54:21,577 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
150
+ 2024-12-10 20:54:22,578 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
151
+ 2024-12-10 20:54:23,162 INFO Thread-12 :598006 [dir_watcher.py:_on_file_modified():288] file/dir modified: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205324/wandb/run-20241210_205330-52nddla7/files/output.log
152
+ 2024-12-10 20:54:23,578 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
153
+ 2024-12-10 20:54:24,578 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
154
+ 2024-12-10 20:54:24,607 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: status_report
155
+ 2024-12-10 20:54:25,162 INFO Thread-12 :598006 [dir_watcher.py:_on_file_modified():288] file/dir modified: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205324/wandb/run-20241210_205330-52nddla7/files/output.log
156
+ 2024-12-10 20:54:25,578 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
157
+ 2024-12-10 20:54:26,578 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
158
+ 2024-12-10 20:54:27,163 INFO Thread-12 :598006 [dir_watcher.py:_on_file_modified():288] file/dir modified: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205324/wandb/run-20241210_205330-52nddla7/files/output.log
159
+ 2024-12-10 20:54:27,579 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
160
+ 2024-12-10 20:54:28,579 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
161
+ 2024-12-10 20:54:29,164 INFO Thread-12 :598006 [dir_watcher.py:_on_file_modified():288] file/dir modified: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205324/wandb/run-20241210_205330-52nddla7/files/output.log
162
+ 2024-12-10 20:54:29,254 DEBUG SenderThread:598006 [sender.py:send():379] send: files
163
+ 2024-12-10 20:54:29,254 INFO SenderThread:598006 [sender.py:_save_file():1454] saving file media/videos/eval/video_0_ff4edecb6a17ea039fbf.mp4 with policy now
164
+ 2024-12-10 20:54:29,256 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: partial_history
165
+ 2024-12-10 20:54:29,257 DEBUG SenderThread:598006 [sender.py:send():379] send: history
166
+ 2024-12-10 20:54:29,257 DEBUG SenderThread:598006 [sender.py:send_request():406] send_request: summary_record
167
+ 2024-12-10 20:54:29,257 INFO SenderThread:598006 [sender.py:_save_file():1454] saving file wandb-summary.json with policy end
168
+ 2024-12-10 20:54:29,258 DEBUG SenderThread:598006 [sender.py:send():379] send: exit
169
+ 2024-12-10 20:54:29,258 INFO SenderThread:598006 [sender.py:send_exit():586] handling exit code: 0
170
+ 2024-12-10 20:54:29,259 INFO SenderThread:598006 [sender.py:send_exit():588] handling runtime: 58
171
+ 2024-12-10 20:54:29,259 INFO SenderThread:598006 [sender.py:_save_file():1454] saving file wandb-summary.json with policy end
172
+ 2024-12-10 20:54:29,259 INFO SenderThread:598006 [sender.py:send_exit():594] send defer
173
+ 2024-12-10 20:54:29,259 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: defer
174
+ 2024-12-10 20:54:29,259 INFO HandlerThread:598006 [handler.py:handle_request_defer():184] handle defer: 0
175
+ 2024-12-10 20:54:29,259 DEBUG SenderThread:598006 [sender.py:send_request():406] send_request: defer
176
+ 2024-12-10 20:54:29,259 INFO SenderThread:598006 [sender.py:send_request_defer():610] handle sender defer: 0
177
+ 2024-12-10 20:54:29,259 INFO SenderThread:598006 [sender.py:transition_state():614] send defer: 1
178
+ 2024-12-10 20:54:29,259 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: defer
179
+ 2024-12-10 20:54:29,259 INFO HandlerThread:598006 [handler.py:handle_request_defer():184] handle defer: 1
180
+ 2024-12-10 20:54:29,260 DEBUG SenderThread:598006 [sender.py:send_request():406] send_request: defer
181
+ 2024-12-10 20:54:29,260 INFO SenderThread:598006 [sender.py:send_request_defer():610] handle sender defer: 1
182
+ 2024-12-10 20:54:29,260 INFO SenderThread:598006 [sender.py:transition_state():614] send defer: 2
183
+ 2024-12-10 20:54:29,260 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: defer
184
+ 2024-12-10 20:54:29,260 INFO HandlerThread:598006 [handler.py:handle_request_defer():184] handle defer: 2
185
+ 2024-12-10 20:54:29,260 INFO HandlerThread:598006 [system_monitor.py:finish():203] Stopping system monitor
186
+ 2024-12-10 20:54:29,260 DEBUG SystemMonitor:598006 [system_monitor.py:_start():172] Starting system metrics aggregation loop
187
+ 2024-12-10 20:54:29,260 DEBUG SystemMonitor:598006 [system_monitor.py:_start():179] Finished system metrics aggregation loop
188
+ 2024-12-10 20:54:29,260 DEBUG SystemMonitor:598006 [system_monitor.py:_start():183] Publishing last batch of metrics
189
+ 2024-12-10 20:54:29,269 INFO HandlerThread:598006 [interfaces.py:finish():200] Joined cpu monitor
190
+ 2024-12-10 20:54:29,269 INFO HandlerThread:598006 [interfaces.py:finish():200] Joined disk monitor
191
+ 2024-12-10 20:54:29,319 INFO HandlerThread:598006 [interfaces.py:finish():200] Joined gpu monitor
192
+ 2024-12-10 20:54:29,319 INFO HandlerThread:598006 [interfaces.py:finish():200] Joined memory monitor
193
+ 2024-12-10 20:54:29,320 INFO HandlerThread:598006 [interfaces.py:finish():200] Joined network monitor
194
+ 2024-12-10 20:54:29,320 DEBUG SenderThread:598006 [sender.py:send_request():406] send_request: defer
195
+ 2024-12-10 20:54:29,320 INFO SenderThread:598006 [sender.py:send_request_defer():610] handle sender defer: 2
196
+ 2024-12-10 20:54:29,320 INFO SenderThread:598006 [sender.py:transition_state():614] send defer: 3
197
+ 2024-12-10 20:54:29,320 DEBUG SenderThread:598006 [sender.py:send():379] send: stats
198
+ 2024-12-10 20:54:29,320 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: defer
199
+ 2024-12-10 20:54:29,320 INFO HandlerThread:598006 [handler.py:handle_request_defer():184] handle defer: 3
200
+ 2024-12-10 20:54:29,321 DEBUG SenderThread:598006 [sender.py:send_request():406] send_request: defer
201
+ 2024-12-10 20:54:29,321 INFO SenderThread:598006 [sender.py:send_request_defer():610] handle sender defer: 3
202
+ 2024-12-10 20:54:29,321 INFO SenderThread:598006 [sender.py:transition_state():614] send defer: 4
203
+ 2024-12-10 20:54:29,321 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: defer
204
+ 2024-12-10 20:54:29,321 INFO HandlerThread:598006 [handler.py:handle_request_defer():184] handle defer: 4
205
+ 2024-12-10 20:54:29,518 INFO wandb-upload_2:598006 [upload_job.py:push():130] Uploaded file /tmp/tmptbytljn2wandb/0kw42edo-media/videos/eval/video_0_ff4edecb6a17ea039fbf.mp4
206
+ 2024-12-10 20:54:30,164 INFO Thread-12 :598006 [dir_watcher.py:_on_file_modified():288] file/dir modified: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205324/wandb/run-20241210_205330-52nddla7/files/output.log
207
+ 2024-12-10 20:54:30,164 INFO Thread-12 :598006 [dir_watcher.py:_on_file_created():271] file/dir created: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205324/wandb/run-20241210_205330-52nddla7/files/media/videos/eval/video_0_ff4edecb6a17ea039fbf.mp4
208
+ 2024-12-10 20:54:30,164 INFO Thread-12 :598006 [dir_watcher.py:_on_file_created():271] file/dir created: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205324/wandb/run-20241210_205330-52nddla7/files/wandb-summary.json
209
+ 2024-12-10 20:54:30,165 INFO Thread-12 :598006 [dir_watcher.py:_on_file_created():271] file/dir created: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205324/wandb/run-20241210_205330-52nddla7/files/media/videos
210
+ 2024-12-10 20:54:30,165 INFO Thread-12 :598006 [dir_watcher.py:_on_file_created():271] file/dir created: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205324/wandb/run-20241210_205330-52nddla7/files/media/videos/eval
211
+ 2024-12-10 20:54:30,165 INFO Thread-12 :598006 [dir_watcher.py:_on_file_created():271] file/dir created: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205324/wandb/run-20241210_205330-52nddla7/files/media
212
+ 2024-12-10 20:54:30,523 INFO wandb-upload_0:598006 [upload_job.py:push():130] Uploaded file /tmp/tmptbytljn2wandb/ks8s91l3-media/videos/eval/video_0_ff4edecb6a17ea039fbf.mp4
213
+ 2024-12-10 20:54:35,581 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: poll_exit
214
+ 2024-12-10 20:54:35,581 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: status_report
215
+ 2024-12-10 20:54:35,581 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: keepalive
216
+ 2024-12-10 20:54:35,581 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: status_report
217
+ 2024-12-10 20:54:35,581 DEBUG SenderThread:598006 [sender.py:send_request():406] send_request: defer
218
+ 2024-12-10 20:54:35,581 INFO SenderThread:598006 [sender.py:send_request_defer():610] handle sender defer: 4
219
+ 2024-12-10 20:54:35,581 INFO SenderThread:598006 [sender.py:transition_state():614] send defer: 5
220
+ 2024-12-10 20:54:35,581 DEBUG SenderThread:598006 [sender.py:send_request():406] send_request: poll_exit
221
+ 2024-12-10 20:54:35,582 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: defer
222
+ 2024-12-10 20:54:35,582 INFO HandlerThread:598006 [handler.py:handle_request_defer():184] handle defer: 5
223
+ 2024-12-10 20:54:35,582 DEBUG SenderThread:598006 [sender.py:send():379] send: summary
224
+ 2024-12-10 20:54:35,582 INFO SenderThread:598006 [sender.py:_save_file():1454] saving file wandb-summary.json with policy end
225
+ 2024-12-10 20:54:35,582 DEBUG SenderThread:598006 [sender.py:send_request():406] send_request: defer
226
+ 2024-12-10 20:54:35,582 INFO SenderThread:598006 [sender.py:send_request_defer():610] handle sender defer: 5
227
+ 2024-12-10 20:54:35,582 INFO SenderThread:598006 [sender.py:transition_state():614] send defer: 6
228
+ 2024-12-10 20:54:35,583 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: defer
229
+ 2024-12-10 20:54:35,583 INFO HandlerThread:598006 [handler.py:handle_request_defer():184] handle defer: 6
230
+ 2024-12-10 20:54:35,583 DEBUG SenderThread:598006 [sender.py:send_request():406] send_request: defer
231
+ 2024-12-10 20:54:35,583 INFO SenderThread:598006 [sender.py:send_request_defer():610] handle sender defer: 6
232
+ 2024-12-10 20:54:35,583 INFO SenderThread:598006 [sender.py:transition_state():614] send defer: 7
233
+ 2024-12-10 20:54:35,583 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: status_report
234
+ 2024-12-10 20:54:35,583 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: defer
235
+ 2024-12-10 20:54:35,583 INFO HandlerThread:598006 [handler.py:handle_request_defer():184] handle defer: 7
236
+ 2024-12-10 20:54:35,583 DEBUG SenderThread:598006 [sender.py:send_request():406] send_request: defer
237
+ 2024-12-10 20:54:35,583 INFO SenderThread:598006 [sender.py:send_request_defer():610] handle sender defer: 7
238
+ 2024-12-10 20:54:36,166 INFO Thread-12 :598006 [dir_watcher.py:_on_file_modified():288] file/dir modified: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205324/wandb/run-20241210_205330-52nddla7/files/wandb-summary.json
239
+ 2024-12-10 20:54:36,260 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: poll_exit
240
+ 2024-12-10 20:54:37,110 INFO SenderThread:598006 [sender.py:transition_state():614] send defer: 8
241
+ 2024-12-10 20:54:37,110 DEBUG SenderThread:598006 [sender.py:send_request():406] send_request: poll_exit
242
+ 2024-12-10 20:54:37,110 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: defer
243
+ 2024-12-10 20:54:37,110 INFO HandlerThread:598006 [handler.py:handle_request_defer():184] handle defer: 8
244
+ 2024-12-10 20:54:37,111 DEBUG SenderThread:598006 [sender.py:send_request():406] send_request: defer
245
+ 2024-12-10 20:54:37,111 INFO SenderThread:598006 [sender.py:send_request_defer():610] handle sender defer: 8
246
+ 2024-12-10 20:54:37,111 INFO SenderThread:598006 [job_builder.py:build():440] Attempting to build job artifact
247
+ 2024-12-10 20:54:37,113 INFO SenderThread:598006 [job_builder.py:_get_source_type():569] is repo sourced job
248
+ 2024-12-10 20:54:37,127 INFO SenderThread:598006 [job_builder.py:build():545] adding wandb-job metadata file
249
+ 2024-12-10 20:54:37,136 INFO SenderThread:598006 [sender.py:transition_state():614] send defer: 9
250
+ 2024-12-10 20:54:37,136 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: defer
251
+ 2024-12-10 20:54:37,136 INFO HandlerThread:598006 [handler.py:handle_request_defer():184] handle defer: 9
252
+ 2024-12-10 20:54:37,137 DEBUG SenderThread:598006 [sender.py:send():379] send: artifact
253
+ 2024-12-10 20:54:37,166 INFO Thread-12 :598006 [dir_watcher.py:_on_file_modified():288] file/dir modified: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205324/wandb/run-20241210_205330-52nddla7/files/output.log
254
+ 2024-12-10 20:54:37,260 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: poll_exit
255
+ 2024-12-10 20:54:37,780 INFO wandb-upload_1:598006 [upload_job.py:push():85] Skipped uploading /home/leonmkim/.local/share/wandb/artifacts/staging/tmp_p_p260o
256
+ 2024-12-10 20:54:37,783 INFO wandb-upload_0:598006 [upload_job.py:push():85] Skipped uploading /tmp/tmppldao70p/wandb-job.json
257
+ 2024-12-10 20:54:37,954 INFO wandb-upload_2:598006 [upload_job.py:push():88] Uploaded file /home/leonmkim/.local/share/wandb/artifacts/staging/tmp4q7vuxho
258
+ 2024-12-10 20:54:38,567 INFO SenderThread:598006 [sender.py:send_artifact():1537] sent artifact job-https___github.com_leonmkim_fish_leon.git_FISH_eval_robot.py - {'id': 'QXJ0aWZhY3Q6MTM2Njc2MTczOA==', 'state': 'PENDING', 'artifactSequence': {'id': 'QXJ0aWZhY3RDb2xsZWN0aW9uOjMxNzk5NjkyNQ==', 'latestArtifact': {'id': 'QXJ0aWZhY3Q6MTM2Njc1OTg1Ng==', 'versionIndex': 336}}}
259
+ 2024-12-10 20:54:38,567 DEBUG SenderThread:598006 [sender.py:send_request():406] send_request: defer
260
+ 2024-12-10 20:54:38,567 INFO SenderThread:598006 [sender.py:send_request_defer():610] handle sender defer: 9
261
+ 2024-12-10 20:54:38,567 INFO SenderThread:598006 [dir_watcher.py:finish():358] shutting down directory watcher
262
+ 2024-12-10 20:54:39,167 INFO SenderThread:598006 [dir_watcher.py:finish():388] scan: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205324/wandb/run-20241210_205330-52nddla7/files
263
+ 2024-12-10 20:54:39,168 INFO SenderThread:598006 [dir_watcher.py:finish():402] scan save: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205324/wandb/run-20241210_205330-52nddla7/files/config.yaml config.yaml
264
+ 2024-12-10 20:54:39,168 INFO SenderThread:598006 [dir_watcher.py:finish():402] scan save: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205324/wandb/run-20241210_205330-52nddla7/files/wandb-summary.json wandb-summary.json
265
+ 2024-12-10 20:54:39,168 INFO SenderThread:598006 [dir_watcher.py:finish():402] scan save: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205324/wandb/run-20241210_205330-52nddla7/files/events.out.tfevents.1733882011.leonmkim-ROG-Strix-G15CS-G15CS.597884.0 events.out.tfevents.1733882011.leonmkim-ROG-Strix-G15CS-G15CS.597884.0
266
+ 2024-12-10 20:54:39,168 INFO SenderThread:598006 [dir_watcher.py:finish():402] scan save: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205324/wandb/run-20241210_205330-52nddla7/files/requirements.txt requirements.txt
267
+ 2024-12-10 20:54:39,168 INFO SenderThread:598006 [dir_watcher.py:finish():402] scan save: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205324/wandb/run-20241210_205330-52nddla7/files/wandb-metadata.json wandb-metadata.json
268
+ 2024-12-10 20:54:39,168 INFO SenderThread:598006 [dir_watcher.py:finish():402] scan save: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205324/wandb/run-20241210_205330-52nddla7/files/output.log output.log
269
+ 2024-12-10 20:54:39,173 INFO SenderThread:598006 [dir_watcher.py:finish():402] scan save: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205324/wandb/run-20241210_205330-52nddla7/files/diff.patch diff.patch
270
+ 2024-12-10 20:54:39,173 INFO SenderThread:598006 [dir_watcher.py:finish():402] scan save: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205324/wandb/run-20241210_205330-52nddla7/files/code/FISH/eval_robot.py code/FISH/eval_robot.py
271
+ 2024-12-10 20:54:39,174 INFO SenderThread:598006 [dir_watcher.py:finish():402] scan save: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205324/wandb/run-20241210_205330-52nddla7/files/media/videos/eval/video_0_ff4edecb6a17ea039fbf.mp4 media/videos/eval/video_0_ff4edecb6a17ea039fbf.mp4
272
+ 2024-12-10 20:54:39,175 INFO SenderThread:598006 [sender.py:transition_state():614] send defer: 10
273
+ 2024-12-10 20:54:39,175 DEBUG SenderThread:598006 [sender.py:send_request():406] send_request: poll_exit
274
+ 2024-12-10 20:54:39,175 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: defer
275
+ 2024-12-10 20:54:39,175 INFO HandlerThread:598006 [handler.py:handle_request_defer():184] handle defer: 10
276
+ 2024-12-10 20:54:39,175 DEBUG SenderThread:598006 [sender.py:send_request():406] send_request: defer
277
+ 2024-12-10 20:54:39,175 INFO SenderThread:598006 [sender.py:send_request_defer():610] handle sender defer: 10
278
+ 2024-12-10 20:54:39,175 INFO SenderThread:598006 [file_pusher.py:finish():169] shutting down file pusher
279
+ 2024-12-10 20:54:39,261 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: poll_exit
280
+ 2024-12-10 20:54:39,261 DEBUG SenderThread:598006 [sender.py:send_request():406] send_request: poll_exit
281
+ 2024-12-10 20:54:39,363 INFO wandb-upload_0:598006 [upload_job.py:push():130] Uploaded file /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205324/wandb/run-20241210_205330-52nddla7/files/wandb-summary.json
282
+ 2024-12-10 20:54:39,397 INFO wandb-upload_2:598006 [upload_job.py:push():130] Uploaded file /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205324/wandb/run-20241210_205330-52nddla7/files/requirements.txt
283
+ 2024-12-10 20:54:39,416 INFO wandb-upload_1:598006 [upload_job.py:push():130] Uploaded file /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205324/wandb/run-20241210_205330-52nddla7/files/config.yaml
284
+ 2024-12-10 20:54:39,447 INFO wandb-upload_3:598006 [upload_job.py:push():130] Uploaded file /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205324/wandb/run-20241210_205330-52nddla7/files/output.log
285
+ 2024-12-10 20:54:39,647 INFO Thread-11 (_thread_body):598006 [sender.py:transition_state():614] send defer: 11
286
+ 2024-12-10 20:54:39,647 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: defer
287
+ 2024-12-10 20:54:39,647 INFO HandlerThread:598006 [handler.py:handle_request_defer():184] handle defer: 11
288
+ 2024-12-10 20:54:39,648 DEBUG SenderThread:598006 [sender.py:send_request():406] send_request: defer
289
+ 2024-12-10 20:54:39,648 INFO SenderThread:598006 [sender.py:send_request_defer():610] handle sender defer: 11
290
+ 2024-12-10 20:54:39,648 INFO SenderThread:598006 [file_pusher.py:join():175] waiting for file pusher
291
+ 2024-12-10 20:54:39,648 INFO SenderThread:598006 [sender.py:transition_state():614] send defer: 12
292
+ 2024-12-10 20:54:39,648 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: defer
293
+ 2024-12-10 20:54:39,648 INFO HandlerThread:598006 [handler.py:handle_request_defer():184] handle defer: 12
294
+ 2024-12-10 20:54:39,648 DEBUG SenderThread:598006 [sender.py:send_request():406] send_request: defer
295
+ 2024-12-10 20:54:39,649 INFO SenderThread:598006 [sender.py:send_request_defer():610] handle sender defer: 12
296
+ 2024-12-10 20:54:39,649 INFO SenderThread:598006 [file_stream.py:finish():601] file stream finish called
297
+ 2024-12-10 20:54:39,709 INFO SenderThread:598006 [file_stream.py:finish():605] file stream finish is done
298
+ 2024-12-10 20:54:39,709 INFO SenderThread:598006 [sender.py:transition_state():614] send defer: 13
299
+ 2024-12-10 20:54:39,709 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: defer
300
+ 2024-12-10 20:54:39,709 INFO HandlerThread:598006 [handler.py:handle_request_defer():184] handle defer: 13
301
+ 2024-12-10 20:54:39,709 DEBUG SenderThread:598006 [sender.py:send_request():406] send_request: defer
302
+ 2024-12-10 20:54:39,709 INFO SenderThread:598006 [sender.py:send_request_defer():610] handle sender defer: 13
303
+ 2024-12-10 20:54:39,709 INFO SenderThread:598006 [sender.py:transition_state():614] send defer: 14
304
+ 2024-12-10 20:54:39,709 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: defer
305
+ 2024-12-10 20:54:39,709 INFO HandlerThread:598006 [handler.py:handle_request_defer():184] handle defer: 14
306
+ 2024-12-10 20:54:39,710 DEBUG SenderThread:598006 [sender.py:send():379] send: final
307
+ 2024-12-10 20:54:39,710 DEBUG SenderThread:598006 [sender.py:send():379] send: footer
308
+ 2024-12-10 20:54:39,710 DEBUG SenderThread:598006 [sender.py:send_request():406] send_request: defer
309
+ 2024-12-10 20:54:39,710 INFO SenderThread:598006 [sender.py:send_request_defer():610] handle sender defer: 14
310
+ 2024-12-10 20:54:39,710 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: poll_exit
311
+ 2024-12-10 20:54:39,710 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: server_info
312
+ 2024-12-10 20:54:39,710 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: poll_exit
313
+ 2024-12-10 20:54:39,711 DEBUG SenderThread:598006 [sender.py:send_request():406] send_request: poll_exit
314
+ 2024-12-10 20:54:39,711 DEBUG SenderThread:598006 [sender.py:send_request():406] send_request: server_info
315
+ 2024-12-10 20:54:39,712 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: get_summary
316
+ 2024-12-10 20:54:39,713 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: sampled_history
317
+ 2024-12-10 20:54:39,713 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: internal_messages
318
+ 2024-12-10 20:54:39,793 DEBUG SenderThread:598006 [sender.py:send_request():406] send_request: poll_exit
319
+ 2024-12-10 20:54:39,794 INFO MainThread:598006 [wandb_run.py:_footer_history_summary_info():4016] rendering history
320
+ 2024-12-10 20:54:39,794 INFO MainThread:598006 [wandb_run.py:_footer_history_summary_info():4048] rendering summary
321
+ 2024-12-10 20:54:39,794 INFO MainThread:598006 [wandb_run.py:_footer_sync_info():3975] logging synced files
322
+ 2024-12-10 20:54:39,794 DEBUG HandlerThread:598006 [handler.py:handle_request():158] handle_request: shutdown
323
+ 2024-12-10 20:54:39,794 INFO HandlerThread:598006 [handler.py:finish():882] shutting down handler
324
+ 2024-12-10 20:54:40,712 INFO WriterThread:598006 [datastore.py:close():296] close: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205324/wandb/run-20241210_205330-52nddla7/run-52nddla7.wandb
325
+ 2024-12-10 20:54:40,794 INFO SenderThread:598006 [sender.py:finish():1615] shutting down sender
326
+ 2024-12-10 20:54:40,794 INFO SenderThread:598006 [file_pusher.py:finish():169] shutting down file pusher
327
+ 2024-12-10 20:54:40,794 INFO SenderThread:598006 [file_pusher.py:join():175] waiting for file pusher
205324/wandb/run-20241210_205330-52nddla7/logs/debug.log ADDED
@@ -0,0 +1,31 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ 2024-12-10 20:53:30,845 INFO MainThread:597884 [wandb_setup.py:_flush():76] Current SDK version is 0.17.5
2
+ 2024-12-10 20:53:30,845 INFO MainThread:597884 [wandb_setup.py:_flush():76] Configure stats pid to 597884
3
+ 2024-12-10 20:53:30,845 INFO MainThread:597884 [wandb_setup.py:_flush():76] Loading settings from /home/leonmkim/.config/wandb/settings
4
+ 2024-12-10 20:53:30,845 INFO MainThread:597884 [wandb_setup.py:_flush():76] Loading settings from /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205324/wandb/settings
5
+ 2024-12-10 20:53:30,845 INFO MainThread:597884 [wandb_setup.py:_flush():76] Loading settings from environment variables: {}
6
+ 2024-12-10 20:53:30,845 INFO MainThread:597884 [wandb_setup.py:_flush():76] Applying setup settings: {'_disable_service': False}
7
+ 2024-12-10 20:53:30,845 INFO MainThread:597884 [wandb_setup.py:_flush():76] Inferring run settings from compute environment: {'program_relpath': 'FISH/eval_robot.py', 'program_abspath': '/home/leonmkim/fish_leon/FISH/eval_robot.py', 'program': '/home/leonmkim/fish_leon/FISH/eval_robot.py'}
8
+ 2024-12-10 20:53:30,845 INFO MainThread:597884 [wandb_setup.py:_flush():76] Applying login settings: {}
9
+ 2024-12-10 20:53:30,845 INFO MainThread:597884 [wandb_init.py:_log_setup():529] Logging user logs to /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205324/wandb/run-20241210_205330-52nddla7/logs/debug.log
10
+ 2024-12-10 20:53:30,845 INFO MainThread:597884 [wandb_init.py:_log_setup():530] Logging internal logs to /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205324/wandb/run-20241210_205330-52nddla7/logs/debug-internal.log
11
+ 2024-12-10 20:53:30,845 INFO MainThread:597884 [wandb_init.py:init():569] calling init triggers
12
+ 2024-12-10 20:53:30,845 INFO MainThread:597884 [wandb_init.py:init():576] wandb.init called with sweep_config: {}
13
+ config: {'root_dir': '/home/leonmkim/fish_leon', 'replay_buffer_size': 150000, 'replay_buffer_num_workers': 2, 'nstep': 3, 'batch_size': 128, 'seed': 0, 'dataset_shuffle_seed': 0, 'device': 'cuda', 'save_video': True, 'save_train_video': True, 'use_tb': True, 'use_wandb': True, 'wandb_run_id': '265626', 'wandb_notes': '265626_req_265827', 'eval': True, 'process_contact_features': True, 'obs_type': 'pixels', 'use_color': True, 'use_depth': True, 'use_masks': True, 'mask_list': ['EE_obj_mask'], 'mask_representation': 'channels', 'crop_hw': [144, 144], 'crop_down_offset': 48, 'color_crop_type': None, 'depth_crop_type': None, 'segmask_crop_type': None, 'add_crop_binary_mask': False, 'add_coord_conv_map': False, 'use_context_color': False, 'use_context_depth': False, 'use_context_segmask': False, 'context_color_crop_type': None, 'context_depth_crop_type': None, 'context_segmask_crop_type': None, 'context_add_crop_binary_mask': False, 'context_add_coord_conv_map': False, 'use_contact_map': False, 'use_sdf_maps': False, 'use_normals_maps': False, 'which_objects': 'both', 'max_contact_prob': 0.1, 'max_depth': 2.0, 'grasped_dtc_max_value': 0.2, 'env_dtc_max_value': 0.4, 'grasped_normals_mask_max_dtc_value': 0.2, 'env_normals_mask_max_dtc_value': 0.4, 'clamp_dtc': True, 'dtc_adaptive_normalization': False, 'mask_normals_within_sdf': True, 'adaptive_normals_mask': True, 'learnable_contact_preprocess_params': True, 'contact_model_name': 'local_multitask_outhd64all_home_crop_h144w144d48_mask_ctxtmask_seed_220979_epoch_9', 'contact_estimation_model_ckpt_path': '~/fish_leon/contact_estimation/artifacts/175604_2/checkpoints/epoch=09-val_loss=0.00.ckpt', 'encoder_type': 'small', 'debug_timestamps': False, 'open_loop': False, 'action_trajectories': True, 'stop_after_action': False, 'interpolation_frequency': 25, 'policy_frequency': 5, 'wait_for_new_camera_frames': True, 'baseline': False, 'train_demo_idxs_list_or_num': -1, 'log_train_every_steps': 25, 'name_of_expert_demo': '64_240x320_hbm_twodim_fps_fix_contact_fix_recovery_fix_annotated_start_idx_5hz_zstd7_EE_pxl_coords_expert_demos_imp_act', 'expert_dataset_dirpath': '/home/leonmkim/fish_leon/FISH/expert_demos/frankagym/FrankaInsertion-v1/64_240x320_hbm_twodim_fps_fix_contact_fix_recovery_fix_annotated_start_idx_5hz_zstd7_EE_pxl_coords_expert_demos_imp_act', 'store_dataset_in_memory': False, 'expert_dataset': '/home/leonmkim/fish_leon/FISH/expert_demos/frankagym/FrankaInsertion-v1/64_240x320_hbm_twodim_fps_fix_contact_fix_recovery_fix_annotated_start_idx_5hz_zstd7_EE_pxl_coords_expert_demos_imp_act/demos.zarr', 'action_key': 'action_trajectory_25hz', 'semantic_demo_grouping_name': 'semantic_demo_grouping.yaml', 'semantic_demo_grouping': '/home/leonmkim/fish_leon/FISH/expert_demos/frankagym/FrankaInsertion-v1/64_240x320_hbm_twodim_fps_fix_contact_fix_recovery_fix_annotated_start_idx_5hz_zstd7_EE_pxl_coords_expert_demos_imp_act/semantic_demo_grouping.yaml', 'include_groups_list': ['hbm_twodim_left_to_right'], 'expert_dataset_config': '/home/leonmkim/fish_leon/FISH/expert_demos/frankagym/FrankaInsertion-v1/64_240x320_hbm_twodim_fps_fix_contact_fix_recovery_fix_annotated_start_idx_5hz_zstd7_EE_pxl_coords_expert_demos_imp_act/demo_config.yaml', 'name_of_valid_demo': '64_240x320_hbm_twodim_fps_fix_contact_fix_recovery_fix_annotated_start_idx_5hz_zstd7_EE_pxl_coords_expert_demos_imp_act', 'valid_dataset_dir': '/home/leonmkim/fish_leon/FISH/expert_demos/frankagym/FrankaInsertion-v1/64_240x320_hbm_twodim_fps_fix_contact_fix_recovery_fix_annotated_start_idx_5hz_zstd7_EE_pxl_coords_expert_demos_imp_act/demos.zarr', 'valid_demo_idxs_list_or_num': None, 'val_num_groups': 0, 'load_bc': True, 'checkpoint_epoch_list': [99, 199, 299, 399, 499, 599, 699, 799, 899, 999, 1249, 1499, 1749, 1999, 2999, 3999, 4999, 5999, 6999, 7999, 8999, 9999], 'snapshot_root_dir': '/mnt/grasp_high_usage/leonmkim/contact_estimation/FISH', 'save_snapshot': True, 'save_last_snapshot': True, 'save_snapshot_when_done': True, 'top_k_checkpoints': 5, 'save_snapshot_link_to_weights_dir': 'deprecated', 'bc_regularize': False, 'bc_weight_type': 'qfilter', 'experiment_dir': './exp_local/frankagym_pixels/FrankaInsertion-v1/265626', 'agent': {'_target_': 'agent.diffusion_policy.DiffusionPolicyAgent', 'name': 'diffusion_policy', 'load_checkpoint': True, 'obs_shape': [5, 240, 320], 'action_shape': [36, 7], 'device': 'cuda', 'n_obs_steps': 1, 'suite_name': 'frankagym', 'obs_type': 'pixels', 'enable_arm': True, 'enable_camera': True, 'use_tb': True, 'desired_image_shape': [13, 180, 240], 'orig_cam_shape': [3, 240, 320], 'config': {'_target_': 'agent.diffusion_policy.DiffusionPolicyAgentConfig', 'compile': False, 'device': 'cuda', 'cam_resize_shape': [13, 180, 240], 'orig_cam_shape': [3, 240, 320], 'policy_cfg': {'_target_': 'lerobot.common.policies.diffusion.configuration_diffusion.DiffusionConfig', 'n_obs_steps': 1, 'horizon': 36, 'n_action_steps': 36, 'output_shapes': {'action': [7]}, 'input_normalization_modes': {'observation.image': 'mean_std', 'observation.state': 'min_max', 'observation.action_history': 'min_max'}, 'output_normalization_modes': {'action': 'min_max'}, 'vision_backbone': 'resnet18', 'pretrained_backbone_weights': None, 'transforms': [{'_target_': 'torchaug.transforms.RandomAffine', 'degrees': [-5, 5], 'translate': [0.05, 0.05], 'batch_transform': True, 'num_chunks': -1, 'batch_inplace': True}, {'_target_': 'torchaug.transforms.RandomColorJitter', 'brightness': 0.3, 'contrast': 0.4, 'saturation': 0.5, 'hue': 0.08, 'batch_transform': True, 'num_chunks': -1, 'batch_inplace': True}], 'use_group_norm': True, 'spatial_softmax_num_keypoints': 32, 'action_history_encoder_config': {'_target_': 'lerobot.common.policies.diffusion.configuration_diffusion.Unet1dEncoderConfig', 'in_channels': 7, 'out_channels': 32, 'history_length': 6, 'kernel_size': 5, 'downsample_kernel_size': 3, 'downsample_stride': 2, 'downsample_padding': 1}, 'down_dims': [256, 512, 1024], 'kernel_size': 5, 'n_groups': 8, 'diffusion_step_embed_dim': 128, 'use_film_scale_modulation': True, 'noise_scheduler_type': 'DDIM', 'beta_schedule': 'squaredcos_cap_v2', 'beta_start': 0.0001, 'beta_end': 0.02, 'prediction_type': 'epsilon', 'clip_sample': True, 'clip_sample_range': 1.0, 'num_train_timesteps': 50, 'num_inference_steps': 10, 'do_mask_loss_for_padding': False, 'input_shapes': {'observation.image': [13, 180, 240], 'context_observation.image': [13, 180, 240], 'observation.state': [8], 'observation.action_history': [7]}}, 'train_cfg': {'_target_': 'utils.TrainConfig', 'lr': 0.0001, 'lr_scheduler': 'cosine', 'lr_warmup_steps': 500, 'adam_betas': [0.95, 0.999], 'adam_eps': 1e-08, 'adam_weight_decay': 1e-06, 'grad_clip_norm': 10, 'offline_steps': 1000000, 'use_amp': True}, 'observation_cfg': {'_target_': 'agent.encoder.VisualFeatureSet', 'use_depth': True, 'use_color': True, 'mask_input_dict': {'_target_': 'agent.encoder.MaskInputDict', 'enable': True, 'representation': 'channels', 'mask_list': ['EE_obj_mask']}, 'crop_input_config': {'_target_': 'agent.encoder.CropInputConfig', 'color_crop_type': None, 'depth_crop_type': None, 'segmask_crop_type': None, 'crop_hw': [144, 144], 'crop_down_offset': 48, 'add_crop_binary_mask': False, 'add_coord_conv_map': False}, 'context_input_config': {'_target_': 'agent.encoder.ContextInputConfig', 'use_color': False, 'use_depth': False, 'mask_input_dict': {'_target_': 'agent.encoder.MaskInputDict', 'enable': False, 'representation': 'channels', 'mask_list': ['EE_obj_mask']}, 'crop_input_config': {'_target_': 'agent.encoder.CropInputConfig', 'color_crop_type': None, 'depth_crop_type': None, 'segmask_crop_type': None, 'crop_hw': [144, 144], 'crop_down_offset': 48, 'add_crop_binary_mask': False, 'add_coord_conv_map': False}}, 'mask_soft_approx_scheduler_config': {'_target_': 'agent.encoder.MaskSoftApproxSchedulerConfig', 'num_steps': 40000, 'initial_value': 10.0, 'final_value': 1000.0, 'interpolation_scheme': 'cosine'}, 'use_contact_map': False, 'use_sdf_maps': False, 'use_normals_maps': False, 'which_objects': 'both', 'grasped_dtc_max_value': 0.2, 'env_dtc_max_value': 0.4, 'grasped_normals_mask_max_dtc_value': 0.2, 'env_normals_mask_max_dtc_value': 0.4, 'clamp_dtc': True, 'max_contact_prob': 0.1, 'mask_normals_within_sdf': True, 'dtc_adaptive_normalization': False, 'adaptive_normals_mask': True, 'max_depth': 2.0, 'image_shape': [13, 180, 240], 'learnable_contact_preprocess_params': True, 'learning_rate': 0.0001, 'weight_decay': 0.0, 'contact_model_name': 'local_multitask_outhd64all_home_crop_h144w144d48_mask_ctxtmask_seed_220979_epoch_9', 'zero_centered': False}, 'policy_frequency': 5, 'interpolation_frequency': 25}}, 'suite': {'suite': 'frankagym', 'name': 'frankagym', 'frame_stack': 1, 'action_repeat': 1, 'discount': 0.99, 'hidden_dim': 1024, 'num_train_frames': 2010, 'num_seed_frames': 260, 'num_train_epochs': 5000, 'validate_every_epochs': 100, 'validate_diffusion_on_action_loss_every_epochs': 500, 'train_eval_diffusion_on_action_loss_every_epochs': 500, 'check_topk_every_epochs': 10, 'save_snapshot_every_epochs': 5000, 'eval_every_frames': 2000, 'num_eval_episodes': 5, 'save_snapshot': True, 'wait_for_user_to_start_episode': True, 'task_make_fn': {'_target_': 'suite.frankagym.make', 'name': 'FrankaInsertion-v1', 'height': 240, 'width': 320, 'frame_stack': 1, 'action_repeat': 1, 'seed': 0, 'enable_arm': True, 'enable_gripper': True, 'start_with_gripper_open': True, 'enable_camera': True, 'path_to_depth_extrinsics': '~/fish_leon/FISH/cfgs/camera_poses/camera_poses_L515/20240904-122305/color_tf_world.npy', 'contact_estimation_model_ckpt_path': '~/fish_leon/contact_estimation/artifacts/175604_2/checkpoints/epoch=09-val_loss=0.00.ckpt', 'x_limit': [0.2, 0.7], 'y_limit': [-0.4, 0.4], 'z_limit': [-0.05, 0.55], 'device': 'cuda', 'interpolation_frequency': 25, 'policy_frequency': 5, 'debug_timestamps': False, 'stop_after_action': False, 'open_loop': False, 'wait_for_new_camera_frames': True, 'action_key': 'action_trajectory_25hz', 'action_trajectory_horizon': 36, 'action_trajectories': True, 'path_to_zarr_dataset': '/home/leonmkim/fish_leon/FISH/expert_demos/frankagym/FrankaInsertion-v1/64_240x320_hbm_twodim_fps_fix_contact_fix_recovery_fix_annotated_start_idx_5hz_zstd7_EE_pxl_coords_expert_demos_imp_act/demos.zarr', 'observation_cfg': {'_target_': 'agent.encoder.VisualFeatureSet', 'use_depth': True, 'use_color': True, 'mask_input_dict': {'_target_': 'agent.encoder.MaskInputDict', 'enable': True, 'representation': 'channels', 'mask_list': ['EE_obj_mask']}, 'crop_input_config': {'_target_': 'agent.encoder.CropInputConfig', 'color_crop_type': None, 'depth_crop_type': None, 'segmask_crop_type': None, 'crop_hw': [144, 144], 'crop_down_offset': 48, 'add_crop_binary_mask': False, 'add_coord_conv_map': False}, 'context_input_config': {'_target_': 'agent.encoder.ContextInputConfig', 'use_color': False, 'use_depth': False, 'mask_input_dict': {'_target_': 'agent.encoder.MaskInputDict', 'enable': False, 'representation': 'channels', 'mask_list': ['EE_obj_mask']}, 'crop_input_config': {'_target_': 'agent.encoder.CropInputConfig', 'color_crop_type': None, 'depth_crop_type': None, 'segmask_crop_type': None, 'crop_hw': [144, 144], 'crop_down_offset': 48, 'add_crop_binary_mask': False, 'add_coord_conv_map': False}}, 'mask_soft_approx_scheduler_config': {'_target_': 'agent.encoder.MaskSoftApproxSchedulerConfig', 'num_steps': 40000, 'initial_value': 10.0, 'final_value': 1000.0, 'interpolation_scheme': 'cosine'}, 'use_contact_map': False, 'use_sdf_maps': False, 'use_normals_maps': False, 'which_objects': 'both', 'grasped_dtc_max_value': 0.2, 'env_dtc_max_value': 0.4, 'grasped_normals_mask_max_dtc_value': 0.2, 'env_normals_mask_max_dtc_value': 0.4, 'clamp_dtc': True, 'max_contact_prob': 0.1, 'mask_normals_within_sdf': True, 'dtc_adaptive_normalization': False, 'adaptive_normals_mask': True, 'max_depth': 2.0, 'image_shape': [13, 180, 240], 'learnable_contact_preprocess_params': True, 'learning_rate': 0.0001, 'weight_decay': 0.0, 'contact_model_name': 'local_multitask_outhd64all_home_crop_h144w144d48_mask_ctxtmask_seed_220979_epoch_9', 'zero_centered': False}, 'agent_policy_cfg': {'_target_': 'agent.diffusion_policy.DiffusionPolicyAgentConfig', 'compile': False, 'device': 'cuda', 'cam_resize_shape': [13, 180, 240], 'orig_cam_shape': [3, 240, 320], 'policy_cfg': {'_target_': 'lerobot.common.policies.diffusion.configuration_diffusion.DiffusionConfig', 'n_obs_steps': 1, 'horizon': 36, 'n_action_steps': 36, 'output_shapes': {'action': [7]}, 'input_normalization_modes': {'observation.image': 'mean_std', 'observation.state': 'min_max', 'observation.action_history': 'min_max'}, 'output_normalization_modes': {'action': 'min_max'}, 'vision_backbone': 'resnet18', 'pretrained_backbone_weights': None, 'transforms': [{'_target_': 'torchaug.transforms.RandomAffine', 'degrees': [-5, 5], 'translate': [0.05, 0.05], 'batch_transform': True, 'num_chunks': -1, 'batch_inplace': True}, {'_target_': 'torchaug.transforms.RandomColorJitter', 'brightness': 0.3, 'contrast': 0.4, 'saturation': 0.5, 'hue': 0.08, 'batch_transform': True, 'num_chunks': -1, 'batch_inplace': True}], 'use_group_norm': True, 'spatial_softmax_num_keypoints': 32, 'action_history_encoder_config': {'_target_': 'lerobot.common.policies.diffusion.configuration_diffusion.Unet1dEncoderConfig', 'in_channels': 7, 'out_channels': 32, 'history_length': 6, 'kernel_size': 5, 'downsample_kernel_size': 3, 'downsample_stride': 2, 'downsample_padding': 1}, 'down_dims': [256, 512, 1024], 'kernel_size': 5, 'n_groups': 8, 'diffusion_step_embed_dim': 128, 'use_film_scale_modulation': True, 'noise_scheduler_type': 'DDIM', 'beta_schedule': 'squaredcos_cap_v2', 'beta_start': 0.0001, 'beta_end': 0.02, 'prediction_type': 'epsilon', 'clip_sample': True, 'clip_sample_range': 1.0, 'num_train_timesteps': 50, 'num_inference_steps': 10, 'do_mask_loss_for_padding': False, 'input_shapes': {'observation.image': [13, 180, 240], 'context_observation.image': [13, 180, 240], 'observation.state': [8], 'observation.action_history': [7]}}, 'train_cfg': {'_target_': 'utils.TrainConfig', 'lr': 0.0001, 'lr_scheduler': 'cosine', 'lr_warmup_steps': 500, 'adam_betas': [0.95, 0.999], 'adam_eps': 1e-08, 'adam_weight_decay': 1e-06, 'grad_clip_norm': 10, 'offline_steps': 1000000, 'use_amp': True}, 'observation_cfg': {'_target_': 'agent.encoder.VisualFeatureSet', 'use_depth': True, 'use_color': True, 'mask_input_dict': {'_target_': 'agent.encoder.MaskInputDict', 'enable': True, 'representation': 'channels', 'mask_list': ['EE_obj_mask']}, 'crop_input_config': {'_target_': 'agent.encoder.CropInputConfig', 'color_crop_type': None, 'depth_crop_type': None, 'segmask_crop_type': None, 'crop_hw': [144, 144], 'crop_down_offset': 48, 'add_crop_binary_mask': False, 'add_coord_conv_map': False}, 'context_input_config': {'_target_': 'agent.encoder.ContextInputConfig', 'use_color': False, 'use_depth': False, 'mask_input_dict': {'_target_': 'agent.encoder.MaskInputDict', 'enable': False, 'representation': 'channels', 'mask_list': ['EE_obj_mask']}, 'crop_input_config': {'_target_': 'agent.encoder.CropInputConfig', 'color_crop_type': None, 'depth_crop_type': None, 'segmask_crop_type': None, 'crop_hw': [144, 144], 'crop_down_offset': 48, 'add_crop_binary_mask': False, 'add_coord_conv_map': False}}, 'mask_soft_approx_scheduler_config': {'_target_': 'agent.encoder.MaskSoftApproxSchedulerConfig', 'num_steps': 40000, 'initial_value': 10.0, 'final_value': 1000.0, 'interpolation_scheme': 'cosine'}, 'use_contact_map': False, 'use_sdf_maps': False, 'use_normals_maps': False, 'which_objects': 'both', 'grasped_dtc_max_value': 0.2, 'env_dtc_max_value': 0.4, 'grasped_normals_mask_max_dtc_value': 0.2, 'env_normals_mask_max_dtc_value': 0.4, 'clamp_dtc': True, 'max_contact_prob': 0.1, 'mask_normals_within_sdf': True, 'dtc_adaptive_normalization': False, 'adaptive_normals_mask': True, 'max_depth': 2.0, 'image_shape': [13, 180, 240], 'learnable_contact_preprocess_params': True, 'learning_rate': 0.0001, 'weight_decay': 0.0, 'contact_model_name': 'local_multitask_outhd64all_home_crop_h144w144d48_mask_ctxtmask_seed_220979_epoch_9', 'zero_centered': False}, 'policy_frequency': 5, 'interpolation_frequency': 25}, 'true_action_history': False}}, 'num_train_frames_bc': 50000, 'num_train_frames_drq': 1100000, 'stddev_schedule_drq': 'linear(1.0,0.1,100000)', 'task_name': 'FrankaInsertion-v1', 'num_train_frames_vinn': 25000, 'num_train_frames_diffusion': 1000000, 'num_train_epochs_bc': 5000, 'num_train_epochs_diffusion': 15000, 'validate_every_epochs_bc': 5, 'validate_every_epochs_diffusion': 250, 'validate_diffusion_on_action_loss_every_epochs': 250, 'train_eval_diffusion_on_action_loss_every_epochs': 250, 'check_topk_every_epochs': 5, 'check_topk_every_epochs_diffusion': 250, 'save_snapshot_every_epochs_diffusion': 1500, 'x_limit': [0.2, 0.7], 'y_limit': [-0.4, 0.4], 'z_limit': [-0.05, 0.55], 'home_displacement': [0.55, 0.0, 0.55, 180.0, 0.0, 0.0], 'enable_gripper': True, 'start_with_gripper_open': True, 'offset_mask': [1, 1, 1, 1, 1, 1], 'path_to_depth_extrinsics': '~/fish_leon/FISH/cfgs/camera_poses/camera_poses_L515/20240904-122305/color_tf_world.npy', 'feature_type': '180x240_1_RGB_D_2.0_msk_channels_EE_obj_mask_acthst_hst6_out32_dwnkrnl3_dwnstrd2_dwnpd1', 'save_buffer': True, 'num_eval': 5, 'random_start': False, 'eval_starts': '/home/leonmkim/fish_leon/FISH/eval_starts/frankagym_pixels/FrankaInsertion-v1', 'num_valid_demos': None, 'load_checkpoint': True, 'true_action_history': False, 'checkpoint_epoch': 12000, 'load_residual_weight': False, 'checkpoint_root_dir': '/home/leonmkim/fish_leon/FISH', 'checkpoint_weight_dir': '/home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626', 'residual_weight': '/home/leonmkim/fish_leon/FISH/weights/frankagym_pixels/FrankaInsertion-v1/weight.pt'}
14
+ 2024-12-10 20:53:30,845 INFO MainThread:597884 [wandb_init.py:init():619] starting backend
15
+ 2024-12-10 20:53:30,845 INFO MainThread:597884 [wandb_init.py:init():623] setting up manager
16
+ 2024-12-10 20:53:30,849 INFO MainThread:597884 [backend.py:_multiprocessing_setup():105] multiprocessing start_methods=fork,spawn,forkserver, using: spawn
17
+ 2024-12-10 20:53:30,850 INFO MainThread:597884 [wandb_init.py:init():631] backend started and connected
18
+ 2024-12-10 20:53:30,862 INFO MainThread:597884 [wandb_init.py:init():720] updated telemetry
19
+ 2024-12-10 20:53:30,871 INFO MainThread:597884 [wandb_init.py:init():753] communicating run to backend with 90.0 second timeout
20
+ 2024-12-10 20:53:31,160 INFO MainThread:597884 [wandb_run.py:_on_init():2435] communicating current version
21
+ 2024-12-10 20:53:31,200 INFO MainThread:597884 [wandb_run.py:_on_init():2444] got version response upgrade_message: "wandb version 0.19.0 is available! To upgrade, please run:\n $ pip install wandb --upgrade"
22
+
23
+ 2024-12-10 20:53:31,201 INFO MainThread:597884 [wandb_init.py:init():804] starting run threads in backend
24
+ 2024-12-10 20:53:31,571 INFO MainThread:597884 [wandb_run.py:_console_start():2413] atexit reg
25
+ 2024-12-10 20:53:31,571 INFO MainThread:597884 [wandb_run.py:_redirect():2255] redirect: wrap_raw
26
+ 2024-12-10 20:53:31,571 INFO MainThread:597884 [wandb_run.py:_redirect():2320] Wrapping output streams.
27
+ 2024-12-10 20:53:31,571 INFO MainThread:597884 [wandb_run.py:_redirect():2345] Redirects installed.
28
+ 2024-12-10 20:53:31,573 INFO MainThread:597884 [wandb_init.py:init():847] run started, returning control to user process
29
+ 2024-12-10 20:53:31,573 INFO MainThread:597884 [wandb_run.py:_tensorboard_callback():1544] tensorboard callback: /home/leonmkim/fish_leon/FISH/exp_local/frankagym_pixels/FrankaInsertion-v1/265626/205324/tb, True
30
+ 2024-12-10 20:53:35,546 INFO MainThread:597884 [wandb_run.py:_config_callback():1382] config_cb None None {'grasped_obj_name': 'hbm', 'left_book_slot': 'twodim'}
31
+ 2024-12-10 20:54:40,795 WARNING MsgRouterThr:597884 [router.py:message_loop():77] message_loop has been closed