sigmundhh commited on
Commit
80900b8
·
verified ·
1 Parent(s): 193fd66

Delete train_2/wandb

Browse files
This view is limited to 50 files because it contains too many changes.   See raw diff
Files changed (50) hide show
  1. train_2/wandb/debug-internal.log +0 -3
  2. train_2/wandb/debug.log +0 -30
  3. train_2/wandb/run-20241120_221447-3pyqkvto_train_2/files/code/train.py +0 -35
  4. train_2/wandb/run-20241120_221447-3pyqkvto_train_2/files/config.yaml +0 -370
  5. train_2/wandb/run-20241120_221447-3pyqkvto_train_2/files/media/videos/test/sim_video_4500000_103613_e2fa34f7dc8e0bbff656.mp4 +0 -0
  6. train_2/wandb/run-20241120_221447-3pyqkvto_train_2/files/media/videos/test/sim_video_4500000_109363_faa42d4a2046d4c5d9c8.mp4 +0 -0
  7. train_2/wandb/run-20241120_221447-3pyqkvto_train_2/files/media/videos/test/sim_video_4500000_113_f3963bae5ae4609d10a2.mp4 +0 -0
  8. train_2/wandb/run-20241120_221447-3pyqkvto_train_2/files/media/videos/test/sim_video_4500000_115113_b911f7d496ad4fce5ef8.mp4 +0 -0
  9. train_2/wandb/run-20241120_221447-3pyqkvto_train_2/files/media/videos/test/sim_video_4500000_11613_24de6120ae322dda5f2d.mp4 +0 -0
  10. train_2/wandb/run-20241120_221447-3pyqkvto_train_2/files/media/videos/test/sim_video_4500000_120863_5a83c15d10d9aa51b351.mp4 +0 -0
  11. train_2/wandb/run-20241120_221447-3pyqkvto_train_2/files/media/videos/test/sim_video_4500000_126613_81977e1914e63a3523fb.mp4 +0 -0
  12. train_2/wandb/run-20241120_221447-3pyqkvto_train_2/files/media/videos/test/sim_video_4500000_132363_4b65f4b1cbf3783ef051.mp4 +0 -0
  13. train_2/wandb/run-20241120_221447-3pyqkvto_train_2/files/media/videos/test/sim_video_4500000_138113_2b4dbd80a5306ae404b2.mp4 +0 -0
  14. train_2/wandb/run-20241120_221447-3pyqkvto_train_2/files/media/videos/test/sim_video_4500000_143863_93c7e98d71d351751b4d.mp4 +0 -0
  15. train_2/wandb/run-20241120_221447-3pyqkvto_train_2/files/media/videos/test/sim_video_4500000_149613_02458c49b5ffb069c85b.mp4 +0 -0
  16. train_2/wandb/run-20241120_221447-3pyqkvto_train_2/files/media/videos/test/sim_video_4500000_155363_80e39274783f782222ce.mp4 +0 -0
  17. train_2/wandb/run-20241120_221447-3pyqkvto_train_2/files/media/videos/test/sim_video_4500000_161113_6175e07e8912557b080e.mp4 +0 -0
  18. train_2/wandb/run-20241120_221447-3pyqkvto_train_2/files/media/videos/test/sim_video_4500000_166863_17b6fcde2e43228b0ae0.mp4 +0 -0
  19. train_2/wandb/run-20241120_221447-3pyqkvto_train_2/files/media/videos/test/sim_video_4500000_172613_08d45d6cb6c1da202ee5.mp4 +0 -0
  20. train_2/wandb/run-20241120_221447-3pyqkvto_train_2/files/media/videos/test/sim_video_4500000_17363_1011d4613e11c8f5ce28.mp4 +0 -0
  21. train_2/wandb/run-20241120_221447-3pyqkvto_train_2/files/media/videos/test/sim_video_4500000_23113_d64538ad4cea3eee38d6.mp4 +0 -0
  22. train_2/wandb/run-20241120_221447-3pyqkvto_train_2/files/media/videos/test/sim_video_4500000_28863_801b5a2439c3794325ce.mp4 +0 -0
  23. train_2/wandb/run-20241120_221447-3pyqkvto_train_2/files/media/videos/test/sim_video_4500000_34613_fd07d2652f04fe9f535b.mp4 +0 -0
  24. train_2/wandb/run-20241120_221447-3pyqkvto_train_2/files/media/videos/test/sim_video_4500000_40363_c79d55d2cde293ef6c51.mp4 +0 -0
  25. train_2/wandb/run-20241120_221447-3pyqkvto_train_2/files/media/videos/test/sim_video_4500000_46113_613774eb467c92eb9fd6.mp4 +0 -0
  26. train_2/wandb/run-20241120_221447-3pyqkvto_train_2/files/media/videos/test/sim_video_4500000_51863_b35a4500dee4b3e984c7.mp4 +0 -0
  27. train_2/wandb/run-20241120_221447-3pyqkvto_train_2/files/media/videos/test/sim_video_4500000_57613_d8c8ef1d9daaaf4c2f4e.mp4 +0 -0
  28. train_2/wandb/run-20241120_221447-3pyqkvto_train_2/files/media/videos/test/sim_video_4500000_5863_71d5488d8cf8db500071.mp4 +0 -0
  29. train_2/wandb/run-20241120_221447-3pyqkvto_train_2/files/media/videos/test/sim_video_4500000_63363_856b80156969e0e1c5ed.mp4 +0 -0
  30. train_2/wandb/run-20241120_221447-3pyqkvto_train_2/files/media/videos/test/sim_video_4500000_69113_19624c4af9dfac27640d.mp4 +0 -0
  31. train_2/wandb/run-20241120_221447-3pyqkvto_train_2/files/media/videos/test/sim_video_4500000_74863_0b73a2c8501ee3dd6bd9.mp4 +0 -0
  32. train_2/wandb/run-20241120_221447-3pyqkvto_train_2/files/media/videos/test/sim_video_4500000_80613_372dc38569916ccfd8be.mp4 +0 -0
  33. train_2/wandb/run-20241120_221447-3pyqkvto_train_2/files/media/videos/test/sim_video_4500000_86363_aac008a51af464dbf659.mp4 +0 -0
  34. train_2/wandb/run-20241120_221447-3pyqkvto_train_2/files/media/videos/test/sim_video_4500000_92113_6a035410fdc493174d5f.mp4 +0 -0
  35. train_2/wandb/run-20241120_221447-3pyqkvto_train_2/files/media/videos/test/sim_video_4500000_97863_1acf92eda2b5cb62fc23.mp4 +0 -0
  36. train_2/wandb/run-20241120_221447-3pyqkvto_train_2/files/media/videos/test/sim_video_4500001_103613_02c34644e0b23c3b3654.mp4 +0 -0
  37. train_2/wandb/run-20241120_221447-3pyqkvto_train_2/files/media/videos/test/sim_video_4500001_109363_286207155c2a2f20a649.mp4 +0 -0
  38. train_2/wandb/run-20241120_221447-3pyqkvto_train_2/files/media/videos/test/sim_video_4500001_113_176b4f198445cbbe2c8b.mp4 +0 -0
  39. train_2/wandb/run-20241120_221447-3pyqkvto_train_2/files/media/videos/test/sim_video_4500001_115113_22aa3c0366b07d9c0049.mp4 +0 -0
  40. train_2/wandb/run-20241120_221447-3pyqkvto_train_2/files/media/videos/test/sim_video_4500001_11613_34f5a25e29bfdf5c11c4.mp4 +0 -0
  41. train_2/wandb/run-20241120_221447-3pyqkvto_train_2/files/media/videos/test/sim_video_4500001_120863_fab59f8e1008ac0def3e.mp4 +0 -0
  42. train_2/wandb/run-20241120_221447-3pyqkvto_train_2/files/media/videos/test/sim_video_4500001_126613_4fba8773aaa19a5463c5.mp4 +0 -0
  43. train_2/wandb/run-20241120_221447-3pyqkvto_train_2/files/media/videos/test/sim_video_4500001_132363_040eddf933e96679874a.mp4 +0 -0
  44. train_2/wandb/run-20241120_221447-3pyqkvto_train_2/files/media/videos/test/sim_video_4500001_138113_15f6c30d270c59407fe3.mp4 +0 -0
  45. train_2/wandb/run-20241120_221447-3pyqkvto_train_2/files/media/videos/test/sim_video_4500001_143863_a3cff04803aef6faf6c7.mp4 +0 -0
  46. train_2/wandb/run-20241120_221447-3pyqkvto_train_2/files/media/videos/test/sim_video_4500001_149613_292d8772b969b6ed21a7.mp4 +0 -0
  47. train_2/wandb/run-20241120_221447-3pyqkvto_train_2/files/media/videos/test/sim_video_4500001_155363_f4a99c9f5df124355253.mp4 +0 -0
  48. train_2/wandb/run-20241120_221447-3pyqkvto_train_2/files/media/videos/test/sim_video_4500001_161113_1e9136903088bb960609.mp4 +0 -0
  49. train_2/wandb/run-20241120_221447-3pyqkvto_train_2/files/media/videos/test/sim_video_4500001_166863_ad3053eae96dbd5840a7.mp4 +0 -0
  50. train_2/wandb/run-20241120_221447-3pyqkvto_train_2/files/media/videos/test/sim_video_4500001_172613_eb2ec32535828a7301f1.mp4 +0 -0
train_2/wandb/debug-internal.log DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:9f697817aa5bd3b391917774cd2c7999f7b7f3b2954acb3e0231b72da318b419
3
- size 91228386
 
 
 
 
train_2/wandb/debug.log DELETED
@@ -1,30 +0,0 @@
1
- 2024-11-20 22:14:47,486 INFO MainThread:3381985 [wandb_setup.py:_flush():76] Current SDK version is 0.17.6
2
- 2024-11-20 22:14:47,487 INFO MainThread:3381985 [wandb_setup.py:_flush():76] Configure stats pid to 3381985
3
- 2024-11-20 22:14:47,487 INFO MainThread:3381985 [wandb_setup.py:_flush():76] Loading settings from /cluster/home/sigmunhh/.config/wandb/settings
4
- 2024-11-20 22:14:47,487 INFO MainThread:3381985 [wandb_setup.py:_flush():76] Loading settings from /tmp/ray/session_2024-11-20_22-13-35_079021_3380951/runtime_resources/working_dir_files/_ray_pkg_a254c17765b6da76/wandb/settings
5
- 2024-11-20 22:14:47,487 INFO MainThread:3381985 [wandb_setup.py:_flush():76] Loading settings from environment variables: {}
6
- 2024-11-20 22:14:47,487 INFO MainThread:3381985 [wandb_setup.py:_flush():76] Applying setup settings: {'_disable_service': False}
7
- 2024-11-20 22:14:47,487 INFO MainThread:3381985 [wandb_setup.py:_flush():76] Inferring run settings from compute environment: {'program_relpath': 'train.py', 'program_abspath': '/tmp/ray/session_2024-11-20_22-13-35_079021_3380951/runtime_resources/working_dir_files/_ray_pkg_a254c17765b6da76/train.py', 'program': '/tmp/ray/session_2024-11-20_22-13-35_079021_3380951/runtime_resources/working_dir_files/_ray_pkg_a254c17765b6da76/train.py'}
8
- 2024-11-20 22:14:47,487 INFO MainThread:3381985 [wandb_setup.py:_flush():76] Applying login settings: {}
9
- 2024-11-20 22:14:47,487 INFO MainThread:3381985 [wandb_init.py:_log_setup():521] Logging user logs to /tmp/ray/session_2024-11-20_22-13-35_079021_3380951/runtime_resources/working_dir_files/_ray_pkg_a254c17765b6da76/data/outputs/2024.11.20/22.13.58_train_tedi_unet_hybrid_lift_image/train_2/wandb/run-20241120_221447-3pyqkvto_train_2/logs/debug.log
10
- 2024-11-20 22:14:47,487 INFO MainThread:3381985 [wandb_init.py:_log_setup():522] Logging internal logs to /tmp/ray/session_2024-11-20_22-13-35_079021_3380951/runtime_resources/working_dir_files/_ray_pkg_a254c17765b6da76/data/outputs/2024.11.20/22.13.58_train_tedi_unet_hybrid_lift_image/train_2/wandb/run-20241120_221447-3pyqkvto_train_2/logs/debug-internal.log
11
- 2024-11-20 22:14:47,487 INFO MainThread:3381985 [wandb_init.py:init():559] calling init triggers
12
- 2024-11-20 22:14:47,488 INFO MainThread:3381985 [wandb_init.py:init():566] wandb.init called with sweep_config: {}
13
- config: {'name': 'train_tedi_unet_hybrid', '_target_': 'diffusion_policy.workspace.train_tedi_unet_hybrid_workspace.TrainTEDIUnetHybridWorkspace', 'task_name': 'lift_image', 'shape_meta': {'obs': {'agentview_image': {'shape': [3, 84, 84], 'type': 'rgb'}, 'robot0_eye_in_hand_image': {'shape': [3, 84, 84], 'type': 'rgb'}, 'robot0_eef_pos': {'shape': [3]}, 'robot0_eef_quat': {'shape': [4]}, 'robot0_gripper_qpos': {'shape': [2]}}, 'action': {'shape': [10]}}, 'exp_name': 'default', 'horizon': 20, 'n_obs_steps': 2, 'n_action_steps': 8, 'n_latency_steps': 0, 'dataset_obs_steps': 2, 'past_action_visible': False, 'keypoint_visible_rate': 1.0, 'obs_as_global_cond': True, 'policy': {'_target_': 'diffusion_policy.policy.tedi_unet_hybrid_image_policy.TEDIUnetHybridImagePolicy', 'shape_meta': {'obs': {'agentview_image': {'shape': [3, 84, 84], 'type': 'rgb'}, 'robot0_eye_in_hand_image': {'shape': [3, 84, 84], 'type': 'rgb'}, 'robot0_eef_pos': {'shape': [3]}, 'robot0_eef_quat': {'shape': [4]}, 'robot0_gripper_qpos': {'shape': [2]}}, 'action': {'shape': [10]}}, 'noise_scheduler': {'_target_': 'diffusion_policy.policy.schedulers.DDPMTEDiScheduler', 'num_train_timesteps': 100, 'beta_start': 0.0001, 'beta_end': 0.02, 'beta_schedule': 'squaredcos_cap_v2', 'variance_type': 'fixed_small', 'clip_sample': True, 'prediction_type': 'epsilon'}, 'horizon': 20, 'n_action_steps': 8, 'n_obs_steps': 2, 'num_inference_steps': 100, 'obs_as_global_cond': True, 'crop_shape': [76, 76], 'diffusion_step_embed_dim': 128, 'down_dims': [512, 1024, 2048], 'kernel_size': 5, 'n_groups': 8, 'cond_predict_scale': True, 'obs_encoder_group_norm': True, 'eval_fixed_crop': True, 'temporally_constant_weight': 0.2, 'temporally_increasing_weight': 0.0, 'temporally_random_weights': 0.0, 'chunk_wise_weight': 0.8, 'buffer_init': 'zero'}, 'ema': {'_target_': 'diffusion_policy.model.diffusion.ema_model.EMAModel', 'update_after_step': 0, 'inv_gamma': 1.0, 'power': 0.75, 'min_value': 0.0, 'max_value': 0.9999}, 'dataloader': {'batch_size': 64, 'num_workers': 16, 'shuffle': True, 'pin_memory': True, 'persistent_workers': True}, 'val_dataloader': {'batch_size': 64, 'num_workers': 16, 'shuffle': False, 'pin_memory': True, 'persistent_workers': True}, 'optimizer': {'_target_': 'torch.optim.AdamW', 'lr': 0.0001, 'betas': [0.95, 0.999], 'eps': 1e-08, 'weight_decay': 1e-06}, 'training': {'device': 'cuda:0', 'seed': 44, 'debug': False, 'resume': True, 'lr_scheduler': 'cosine', 'lr_warmup_steps': 500, 'num_epochs': 1501, 'gradient_accumulate_every': 1, 'use_ema': True, 'rollout_every': 50, 'checkpoint_every': 50, 'val_every': 1, 'sample_every': 5, 'max_train_steps': None, 'max_val_steps': None, 'tqdm_interval_sec': 1.0}, 'logging': {'project': 'diffusion_policy_debug', 'resume': True, 'mode': 'online', 'name': '2024.11.20-22.13.58_train_tedi_unet_hybrid_lift_image_train_2', 'tags': ['train_tedi_unet_hybrid', 'lift_image', 'default'], 'id': '3pyqkvto_train_2', 'group': '3pyqkvto'}, 'checkpoint': {'topk': {'monitor_key': 'test_mean_score', 'mode': 'max', 'k': 1, 'format_str': 'epoch={epoch:04d}-test_mean_score={test_mean_score:.3f}.ckpt'}, 'save_last_ckpt': False, 'save_last_snapshot': False}, 'multi_run': {'run_dir': 'data/outputs/2024.11.20/22.14.21_train_tedi_unet_hybrid_lift_image', 'wandb_name_base': '2024.11.20-22.14.21_train_tedi_unet_hybrid_lift_image'}, 'task': {'name': 'lift_image', 'shape_meta': {'obs': {'agentview_image': {'shape': [3, 84, 84], 'type': 'rgb'}, 'robot0_eye_in_hand_image': {'shape': [3, 84, 84], 'type': 'rgb'}, 'robot0_eef_pos': {'shape': [3]}, 'robot0_eef_quat': {'shape': [4]}, 'robot0_gripper_qpos': {'shape': [2]}}, 'action': {'shape': [10]}}, 'task_name': 'lift', 'dataset_type': 'ph', 'dataset_path': 'data/robomimic/datasets/lift/ph/image_abs.hdf5', 'abs_action': True, 'env_runner': {'_target_': 'diffusion_policy.env_runner.robomimic_image_runner.RobomimicImageRunner', 'dataset_path': 'data/robomimic/datasets/lift/ph/image_abs.hdf5', 'shape_meta': {'obs': {'agentview_image': {'shape': [3, 84, 84], 'type': 'rgb'}, 'robot0_eye_in_hand_image': {'shape': [3, 84, 84], 'type': 'rgb'}, 'robot0_eef_pos': {'shape': [3]}, 'robot0_eef_quat': {'shape': [4]}, 'robot0_gripper_qpos': {'shape': [2]}}, 'action': {'shape': [10]}}, 'n_train': 6, 'n_train_vis': 2, 'train_start_idx': 0, 'n_test': 50, 'n_test_vis': 4, 'test_start_seed': 4500000, 'max_steps': 400, 'n_obs_steps': 2, 'n_action_steps': 8, 'render_obs_key': 'agentview_image', 'fps': 10, 'crf': 22, 'past_action': False, 'abs_action': True, 'tqdm_interval_sec': 1.0, 'n_envs': 28}, 'dataset': {'_target_': 'diffusion_policy.dataset.robomimic_replay_image_dataset.RobomimicReplayImageDataset', 'shape_meta': {'obs': {'agentview_image': {'shape': [3, 84, 84], 'type': 'rgb'}, 'robot0_eye_in_hand_image': {'shape': [3, 84, 84], 'type': 'rgb'}, 'robot0_eef_pos': {'shape': [3]}, 'robot0_eef_quat': {'shape': [4]}, 'robot0_gripper_qpos': {'shape': [2]}}, 'action': {'shape': [10]}}, 'dataset_path': 'data/robomimic/datasets/lift/ph/image_abs.hdf5', 'horizon': 20, 'pad_before': 1, 'pad_after': 7, 'n_obs_steps': 2, 'abs_action': True, 'rotation_rep': 'rotation_6d', 'use_legacy_normalizer': False, 'use_cache': True, 'seed': 42, 'val_ratio': 0.02}}}
14
- 2024-11-20 22:14:47,488 INFO MainThread:3381985 [wandb_init.py:init():609] starting backend
15
- 2024-11-20 22:14:47,488 INFO MainThread:3381985 [wandb_init.py:init():613] setting up manager
16
- 2024-11-20 22:14:47,491 INFO MainThread:3381985 [backend.py:_multiprocessing_setup():105] multiprocessing start_methods=fork,spawn,forkserver, using: spawn
17
- 2024-11-20 22:14:47,519 INFO MainThread:3381985 [wandb_init.py:init():621] backend started and connected
18
- 2024-11-20 22:14:47,524 INFO MainThread:3381985 [wandb_init.py:init():716] updated telemetry
19
- 2024-11-20 22:14:47,524 INFO MainThread:3381985 [wandb_init.py:init():749] communicating run to backend with 90.0 second timeout
20
- 2024-11-20 22:14:48,998 INFO MainThread:3381985 [wandb_run.py:_on_init():2438] communicating current version
21
- 2024-11-20 22:14:49,096 INFO MainThread:3381985 [wandb_run.py:_on_init():2447] got version response upgrade_message: "wandb version 0.18.7 is available! To upgrade, please run:\n $ pip install wandb --upgrade"
22
-
23
- 2024-11-20 22:14:49,096 INFO MainThread:3381985 [wandb_init.py:init():800] starting run threads in backend
24
- 2024-11-20 22:14:49,276 INFO MainThread:3381985 [wandb_run.py:_console_start():2416] atexit reg
25
- 2024-11-20 22:14:49,276 INFO MainThread:3381985 [wandb_run.py:_redirect():2258] redirect: wrap_raw
26
- 2024-11-20 22:14:49,276 INFO MainThread:3381985 [wandb_run.py:_redirect():2323] Wrapping output streams.
27
- 2024-11-20 22:14:49,276 INFO MainThread:3381985 [wandb_run.py:_redirect():2348] Redirects installed.
28
- 2024-11-20 22:14:49,279 INFO MainThread:3381985 [wandb_init.py:init():843] run started, returning control to user process
29
- 2024-11-20 22:14:49,279 INFO MainThread:3381985 [wandb_run.py:_config_callback():1385] config_cb None None {'output_dir': '/tmp/ray/session_2024-11-20_22-13-35_079021_3380951/runtime_resources/working_dir_files/_ray_pkg_a254c17765b6da76/data/outputs/2024.11.20/22.13.58_train_tedi_unet_hybrid_lift_image/train_2'}
30
- 2024-11-21 05:12:39,289 WARNING MsgRouterThr:3381985 [router.py:message_loop():77] message_loop has been closed
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
train_2/wandb/run-20241120_221447-3pyqkvto_train_2/files/code/train.py DELETED
@@ -1,35 +0,0 @@
1
- """
2
- Usage:
3
- Training:
4
- python train.py --config-name=train_diffusion_lowdim_workspace
5
- """
6
-
7
- import sys
8
- # use line-buffering for both stdout and stderr
9
- sys.stdout = open(sys.stdout.fileno(), mode='w', buffering=1)
10
- sys.stderr = open(sys.stderr.fileno(), mode='w', buffering=1)
11
-
12
- import hydra
13
- from omegaconf import OmegaConf
14
- import pathlib
15
- from diffusion_policy.workspace.base_workspace import BaseWorkspace
16
-
17
- # allows arbitrary python code execution in configs using the ${eval:''} resolver
18
- OmegaConf.register_new_resolver("eval", eval, replace=True)
19
-
20
- @hydra.main(
21
- version_base=None,
22
- config_path=str(pathlib.Path(__file__).parent.joinpath(
23
- 'diffusion_policy','config'))
24
- )
25
- def main(cfg: OmegaConf):
26
- # resolve immediately so all the ${now:} resolvers
27
- # will use the same time.
28
- OmegaConf.resolve(cfg)
29
-
30
- cls = hydra.utils.get_class(cfg._target_)
31
- workspace: BaseWorkspace = cls(cfg)
32
- workspace.run()
33
-
34
- if __name__ == "__main__":
35
- main()
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
train_2/wandb/run-20241120_221447-3pyqkvto_train_2/files/config.yaml DELETED
@@ -1,370 +0,0 @@
1
- wandb_version: 1
2
-
3
- name:
4
- desc: null
5
- value: train_tedi_unet_hybrid
6
- _target_:
7
- desc: null
8
- value: diffusion_policy.workspace.train_tedi_unet_hybrid_workspace.TrainTEDIUnetHybridWorkspace
9
- task_name:
10
- desc: null
11
- value: lift_image
12
- shape_meta:
13
- desc: null
14
- value:
15
- obs:
16
- agentview_image:
17
- shape:
18
- - 3
19
- - 84
20
- - 84
21
- type: rgb
22
- robot0_eye_in_hand_image:
23
- shape:
24
- - 3
25
- - 84
26
- - 84
27
- type: rgb
28
- robot0_eef_pos:
29
- shape:
30
- - 3
31
- robot0_eef_quat:
32
- shape:
33
- - 4
34
- robot0_gripper_qpos:
35
- shape:
36
- - 2
37
- action:
38
- shape:
39
- - 10
40
- exp_name:
41
- desc: null
42
- value: default
43
- horizon:
44
- desc: null
45
- value: 20
46
- n_obs_steps:
47
- desc: null
48
- value: 2
49
- n_action_steps:
50
- desc: null
51
- value: 8
52
- n_latency_steps:
53
- desc: null
54
- value: 0
55
- dataset_obs_steps:
56
- desc: null
57
- value: 2
58
- past_action_visible:
59
- desc: null
60
- value: false
61
- keypoint_visible_rate:
62
- desc: null
63
- value: 1.0
64
- obs_as_global_cond:
65
- desc: null
66
- value: true
67
- policy:
68
- desc: null
69
- value:
70
- _target_: diffusion_policy.policy.tedi_unet_hybrid_image_policy.TEDIUnetHybridImagePolicy
71
- shape_meta:
72
- obs:
73
- agentview_image:
74
- shape:
75
- - 3
76
- - 84
77
- - 84
78
- type: rgb
79
- robot0_eye_in_hand_image:
80
- shape:
81
- - 3
82
- - 84
83
- - 84
84
- type: rgb
85
- robot0_eef_pos:
86
- shape:
87
- - 3
88
- robot0_eef_quat:
89
- shape:
90
- - 4
91
- robot0_gripper_qpos:
92
- shape:
93
- - 2
94
- action:
95
- shape:
96
- - 10
97
- noise_scheduler:
98
- _target_: diffusion_policy.policy.schedulers.DDPMTEDiScheduler
99
- num_train_timesteps: 100
100
- beta_start: 0.0001
101
- beta_end: 0.02
102
- beta_schedule: squaredcos_cap_v2
103
- variance_type: fixed_small
104
- clip_sample: true
105
- prediction_type: epsilon
106
- horizon: 20
107
- n_action_steps: 8
108
- n_obs_steps: 2
109
- num_inference_steps: 100
110
- obs_as_global_cond: true
111
- crop_shape:
112
- - 76
113
- - 76
114
- diffusion_step_embed_dim: 128
115
- down_dims:
116
- - 512
117
- - 1024
118
- - 2048
119
- kernel_size: 5
120
- n_groups: 8
121
- cond_predict_scale: true
122
- obs_encoder_group_norm: true
123
- eval_fixed_crop: true
124
- temporally_constant_weight: 0.2
125
- temporally_increasing_weight: 0.0
126
- temporally_random_weights: 0.0
127
- chunk_wise_weight: 0.8
128
- buffer_init: zero
129
- ema:
130
- desc: null
131
- value:
132
- _target_: diffusion_policy.model.diffusion.ema_model.EMAModel
133
- update_after_step: 0
134
- inv_gamma: 1.0
135
- power: 0.75
136
- min_value: 0.0
137
- max_value: 0.9999
138
- dataloader:
139
- desc: null
140
- value:
141
- batch_size: 64
142
- num_workers: 16
143
- shuffle: true
144
- pin_memory: true
145
- persistent_workers: true
146
- val_dataloader:
147
- desc: null
148
- value:
149
- batch_size: 64
150
- num_workers: 16
151
- shuffle: false
152
- pin_memory: true
153
- persistent_workers: true
154
- optimizer:
155
- desc: null
156
- value:
157
- _target_: torch.optim.AdamW
158
- lr: 0.0001
159
- betas:
160
- - 0.95
161
- - 0.999
162
- eps: 1.0e-08
163
- weight_decay: 1.0e-06
164
- training:
165
- desc: null
166
- value:
167
- device: cuda:0
168
- seed: 44
169
- debug: false
170
- resume: true
171
- lr_scheduler: cosine
172
- lr_warmup_steps: 500
173
- num_epochs: 1501
174
- gradient_accumulate_every: 1
175
- use_ema: true
176
- rollout_every: 50
177
- checkpoint_every: 50
178
- val_every: 1
179
- sample_every: 5
180
- max_train_steps: null
181
- max_val_steps: null
182
- tqdm_interval_sec: 1.0
183
- logging:
184
- desc: null
185
- value:
186
- project: diffusion_policy_debug
187
- resume: true
188
- mode: online
189
- name: 2024.11.20-22.13.58_train_tedi_unet_hybrid_lift_image_train_2
190
- tags:
191
- - train_tedi_unet_hybrid
192
- - lift_image
193
- - default
194
- id: 3pyqkvto_train_2
195
- group: 3pyqkvto
196
- checkpoint:
197
- desc: null
198
- value:
199
- topk:
200
- monitor_key: test_mean_score
201
- mode: max
202
- k: 1
203
- format_str: epoch={epoch:04d}-test_mean_score={test_mean_score:.3f}.ckpt
204
- save_last_ckpt: false
205
- save_last_snapshot: false
206
- multi_run:
207
- desc: null
208
- value:
209
- run_dir: data/outputs/2024.11.20/22.14.21_train_tedi_unet_hybrid_lift_image
210
- wandb_name_base: 2024.11.20-22.14.21_train_tedi_unet_hybrid_lift_image
211
- task:
212
- desc: null
213
- value:
214
- name: lift_image
215
- shape_meta:
216
- obs:
217
- agentview_image:
218
- shape:
219
- - 3
220
- - 84
221
- - 84
222
- type: rgb
223
- robot0_eye_in_hand_image:
224
- shape:
225
- - 3
226
- - 84
227
- - 84
228
- type: rgb
229
- robot0_eef_pos:
230
- shape:
231
- - 3
232
- robot0_eef_quat:
233
- shape:
234
- - 4
235
- robot0_gripper_qpos:
236
- shape:
237
- - 2
238
- action:
239
- shape:
240
- - 10
241
- task_name: lift
242
- dataset_type: ph
243
- dataset_path: data/robomimic/datasets/lift/ph/image_abs.hdf5
244
- abs_action: true
245
- env_runner:
246
- _target_: diffusion_policy.env_runner.robomimic_image_runner.RobomimicImageRunner
247
- dataset_path: data/robomimic/datasets/lift/ph/image_abs.hdf5
248
- shape_meta:
249
- obs:
250
- agentview_image:
251
- shape:
252
- - 3
253
- - 84
254
- - 84
255
- type: rgb
256
- robot0_eye_in_hand_image:
257
- shape:
258
- - 3
259
- - 84
260
- - 84
261
- type: rgb
262
- robot0_eef_pos:
263
- shape:
264
- - 3
265
- robot0_eef_quat:
266
- shape:
267
- - 4
268
- robot0_gripper_qpos:
269
- shape:
270
- - 2
271
- action:
272
- shape:
273
- - 10
274
- n_train: 6
275
- n_train_vis: 2
276
- train_start_idx: 0
277
- n_test: 50
278
- n_test_vis: 4
279
- test_start_seed: 4500000
280
- max_steps: 400
281
- n_obs_steps: 2
282
- n_action_steps: 8
283
- render_obs_key: agentview_image
284
- fps: 10
285
- crf: 22
286
- past_action: false
287
- abs_action: true
288
- tqdm_interval_sec: 1.0
289
- n_envs: 28
290
- dataset:
291
- _target_: diffusion_policy.dataset.robomimic_replay_image_dataset.RobomimicReplayImageDataset
292
- shape_meta:
293
- obs:
294
- agentview_image:
295
- shape:
296
- - 3
297
- - 84
298
- - 84
299
- type: rgb
300
- robot0_eye_in_hand_image:
301
- shape:
302
- - 3
303
- - 84
304
- - 84
305
- type: rgb
306
- robot0_eef_pos:
307
- shape:
308
- - 3
309
- robot0_eef_quat:
310
- shape:
311
- - 4
312
- robot0_gripper_qpos:
313
- shape:
314
- - 2
315
- action:
316
- shape:
317
- - 10
318
- dataset_path: data/robomimic/datasets/lift/ph/image_abs.hdf5
319
- horizon: 20
320
- pad_before: 1
321
- pad_after: 7
322
- n_obs_steps: 2
323
- abs_action: true
324
- rotation_rep: rotation_6d
325
- use_legacy_normalizer: false
326
- use_cache: true
327
- seed: 42
328
- val_ratio: 0.02
329
- _wandb:
330
- desc: null
331
- value:
332
- code_path: code/train.py
333
- python_version: 3.9.15
334
- cli_version: 0.17.6
335
- framework: torch
336
- is_jupyter_run: false
337
- is_kaggle_kernel: false
338
- start_time: 1732137287
339
- t:
340
- 1:
341
- - 1
342
- - 41
343
- - 49
344
- - 50
345
- - 55
346
- - 71
347
- - 83
348
- 2:
349
- - 1
350
- - 41
351
- - 49
352
- - 50
353
- - 55
354
- - 71
355
- - 83
356
- 3:
357
- - 13
358
- - 14
359
- - 15
360
- - 16
361
- - 23
362
- - 61
363
- 4: 3.9.15
364
- 5: 0.17.6
365
- 8:
366
- - 5
367
- 13: linux-x86_64
368
- output_dir:
369
- desc: null
370
- value: /tmp/ray/session_2024-11-20_22-13-35_079021_3380951/runtime_resources/working_dir_files/_ray_pkg_a254c17765b6da76/data/outputs/2024.11.20/22.13.58_train_tedi_unet_hybrid_lift_image/train_2
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
train_2/wandb/run-20241120_221447-3pyqkvto_train_2/files/media/videos/test/sim_video_4500000_103613_e2fa34f7dc8e0bbff656.mp4 DELETED
Binary file (48 kB)
 
train_2/wandb/run-20241120_221447-3pyqkvto_train_2/files/media/videos/test/sim_video_4500000_109363_faa42d4a2046d4c5d9c8.mp4 DELETED
Binary file (48.5 kB)
 
train_2/wandb/run-20241120_221447-3pyqkvto_train_2/files/media/videos/test/sim_video_4500000_113_f3963bae5ae4609d10a2.mp4 DELETED
Binary file (79.6 kB)
 
train_2/wandb/run-20241120_221447-3pyqkvto_train_2/files/media/videos/test/sim_video_4500000_115113_b911f7d496ad4fce5ef8.mp4 DELETED
Binary file (47.4 kB)
 
train_2/wandb/run-20241120_221447-3pyqkvto_train_2/files/media/videos/test/sim_video_4500000_11613_24de6120ae322dda5f2d.mp4 DELETED
Binary file (53.9 kB)
 
train_2/wandb/run-20241120_221447-3pyqkvto_train_2/files/media/videos/test/sim_video_4500000_120863_5a83c15d10d9aa51b351.mp4 DELETED
Binary file (47 kB)
 
train_2/wandb/run-20241120_221447-3pyqkvto_train_2/files/media/videos/test/sim_video_4500000_126613_81977e1914e63a3523fb.mp4 DELETED
Binary file (46.1 kB)
 
train_2/wandb/run-20241120_221447-3pyqkvto_train_2/files/media/videos/test/sim_video_4500000_132363_4b65f4b1cbf3783ef051.mp4 DELETED
Binary file (45.5 kB)
 
train_2/wandb/run-20241120_221447-3pyqkvto_train_2/files/media/videos/test/sim_video_4500000_138113_2b4dbd80a5306ae404b2.mp4 DELETED
Binary file (45.5 kB)
 
train_2/wandb/run-20241120_221447-3pyqkvto_train_2/files/media/videos/test/sim_video_4500000_143863_93c7e98d71d351751b4d.mp4 DELETED
Binary file (45.5 kB)
 
train_2/wandb/run-20241120_221447-3pyqkvto_train_2/files/media/videos/test/sim_video_4500000_149613_02458c49b5ffb069c85b.mp4 DELETED
Binary file (44.9 kB)
 
train_2/wandb/run-20241120_221447-3pyqkvto_train_2/files/media/videos/test/sim_video_4500000_155363_80e39274783f782222ce.mp4 DELETED
Binary file (45.3 kB)
 
train_2/wandb/run-20241120_221447-3pyqkvto_train_2/files/media/videos/test/sim_video_4500000_161113_6175e07e8912557b080e.mp4 DELETED
Binary file (44.5 kB)
 
train_2/wandb/run-20241120_221447-3pyqkvto_train_2/files/media/videos/test/sim_video_4500000_166863_17b6fcde2e43228b0ae0.mp4 DELETED
Binary file (45 kB)
 
train_2/wandb/run-20241120_221447-3pyqkvto_train_2/files/media/videos/test/sim_video_4500000_172613_08d45d6cb6c1da202ee5.mp4 DELETED
Binary file (45.5 kB)
 
train_2/wandb/run-20241120_221447-3pyqkvto_train_2/files/media/videos/test/sim_video_4500000_17363_1011d4613e11c8f5ce28.mp4 DELETED
Binary file (50.2 kB)
 
train_2/wandb/run-20241120_221447-3pyqkvto_train_2/files/media/videos/test/sim_video_4500000_23113_d64538ad4cea3eee38d6.mp4 DELETED
Binary file (49.3 kB)
 
train_2/wandb/run-20241120_221447-3pyqkvto_train_2/files/media/videos/test/sim_video_4500000_28863_801b5a2439c3794325ce.mp4 DELETED
Binary file (49.9 kB)
 
train_2/wandb/run-20241120_221447-3pyqkvto_train_2/files/media/videos/test/sim_video_4500000_34613_fd07d2652f04fe9f535b.mp4 DELETED
Binary file (51.6 kB)
 
train_2/wandb/run-20241120_221447-3pyqkvto_train_2/files/media/videos/test/sim_video_4500000_40363_c79d55d2cde293ef6c51.mp4 DELETED
Binary file (52.9 kB)
 
train_2/wandb/run-20241120_221447-3pyqkvto_train_2/files/media/videos/test/sim_video_4500000_46113_613774eb467c92eb9fd6.mp4 DELETED
Binary file (50.3 kB)
 
train_2/wandb/run-20241120_221447-3pyqkvto_train_2/files/media/videos/test/sim_video_4500000_51863_b35a4500dee4b3e984c7.mp4 DELETED
Binary file (49.3 kB)
 
train_2/wandb/run-20241120_221447-3pyqkvto_train_2/files/media/videos/test/sim_video_4500000_57613_d8c8ef1d9daaaf4c2f4e.mp4 DELETED
Binary file (50.9 kB)
 
train_2/wandb/run-20241120_221447-3pyqkvto_train_2/files/media/videos/test/sim_video_4500000_5863_71d5488d8cf8db500071.mp4 DELETED
Binary file (56.4 kB)
 
train_2/wandb/run-20241120_221447-3pyqkvto_train_2/files/media/videos/test/sim_video_4500000_63363_856b80156969e0e1c5ed.mp4 DELETED
Binary file (50.1 kB)
 
train_2/wandb/run-20241120_221447-3pyqkvto_train_2/files/media/videos/test/sim_video_4500000_69113_19624c4af9dfac27640d.mp4 DELETED
Binary file (50.5 kB)
 
train_2/wandb/run-20241120_221447-3pyqkvto_train_2/files/media/videos/test/sim_video_4500000_74863_0b73a2c8501ee3dd6bd9.mp4 DELETED
Binary file (50 kB)
 
train_2/wandb/run-20241120_221447-3pyqkvto_train_2/files/media/videos/test/sim_video_4500000_80613_372dc38569916ccfd8be.mp4 DELETED
Binary file (50.6 kB)
 
train_2/wandb/run-20241120_221447-3pyqkvto_train_2/files/media/videos/test/sim_video_4500000_86363_aac008a51af464dbf659.mp4 DELETED
Binary file (49.2 kB)
 
train_2/wandb/run-20241120_221447-3pyqkvto_train_2/files/media/videos/test/sim_video_4500000_92113_6a035410fdc493174d5f.mp4 DELETED
Binary file (48.6 kB)
 
train_2/wandb/run-20241120_221447-3pyqkvto_train_2/files/media/videos/test/sim_video_4500000_97863_1acf92eda2b5cb62fc23.mp4 DELETED
Binary file (47.5 kB)
 
train_2/wandb/run-20241120_221447-3pyqkvto_train_2/files/media/videos/test/sim_video_4500001_103613_02c34644e0b23c3b3654.mp4 DELETED
Binary file (52.6 kB)
 
train_2/wandb/run-20241120_221447-3pyqkvto_train_2/files/media/videos/test/sim_video_4500001_109363_286207155c2a2f20a649.mp4 DELETED
Binary file (51.9 kB)
 
train_2/wandb/run-20241120_221447-3pyqkvto_train_2/files/media/videos/test/sim_video_4500001_113_176b4f198445cbbe2c8b.mp4 DELETED
Binary file (71.6 kB)
 
train_2/wandb/run-20241120_221447-3pyqkvto_train_2/files/media/videos/test/sim_video_4500001_115113_22aa3c0366b07d9c0049.mp4 DELETED
Binary file (52 kB)
 
train_2/wandb/run-20241120_221447-3pyqkvto_train_2/files/media/videos/test/sim_video_4500001_11613_34f5a25e29bfdf5c11c4.mp4 DELETED
Binary file (54.1 kB)
 
train_2/wandb/run-20241120_221447-3pyqkvto_train_2/files/media/videos/test/sim_video_4500001_120863_fab59f8e1008ac0def3e.mp4 DELETED
Binary file (51.4 kB)
 
train_2/wandb/run-20241120_221447-3pyqkvto_train_2/files/media/videos/test/sim_video_4500001_126613_4fba8773aaa19a5463c5.mp4 DELETED
Binary file (51.2 kB)
 
train_2/wandb/run-20241120_221447-3pyqkvto_train_2/files/media/videos/test/sim_video_4500001_132363_040eddf933e96679874a.mp4 DELETED
Binary file (51.6 kB)
 
train_2/wandb/run-20241120_221447-3pyqkvto_train_2/files/media/videos/test/sim_video_4500001_138113_15f6c30d270c59407fe3.mp4 DELETED
Binary file (51.6 kB)
 
train_2/wandb/run-20241120_221447-3pyqkvto_train_2/files/media/videos/test/sim_video_4500001_143863_a3cff04803aef6faf6c7.mp4 DELETED
Binary file (51 kB)
 
train_2/wandb/run-20241120_221447-3pyqkvto_train_2/files/media/videos/test/sim_video_4500001_149613_292d8772b969b6ed21a7.mp4 DELETED
Binary file (50.4 kB)
 
train_2/wandb/run-20241120_221447-3pyqkvto_train_2/files/media/videos/test/sim_video_4500001_155363_f4a99c9f5df124355253.mp4 DELETED
Binary file (51.2 kB)
 
train_2/wandb/run-20241120_221447-3pyqkvto_train_2/files/media/videos/test/sim_video_4500001_161113_1e9136903088bb960609.mp4 DELETED
Binary file (51.7 kB)
 
train_2/wandb/run-20241120_221447-3pyqkvto_train_2/files/media/videos/test/sim_video_4500001_166863_ad3053eae96dbd5840a7.mp4 DELETED
Binary file (51.5 kB)
 
train_2/wandb/run-20241120_221447-3pyqkvto_train_2/files/media/videos/test/sim_video_4500001_172613_eb2ec32535828a7301f1.mp4 DELETED
Binary file (51.1 kB)