Commit Graph

  • d16f6a93b3 Merge remote-tracking branch 'upstream/main' into user/alexander-soare/multistep_policy_and_serial_env Alexander Soare 2024-03-20 09:01:45 +0000
  • 52e149fbfd Only save video frames in first rollout Alexander Soare 2024-03-20 08:32:11 +0000
  • 4f1955edfd Clear action queue when environment is reset Alexander Soare 2024-03-20 08:31:06 +0000
  • c5010fee9a fix seeding Alexander Soare 2024-03-20 08:20:56 +0000
  • 18fa88475b Move reset_warning_issued flag to class attribute Alexander Soare 2024-03-20 08:09:38 +0000
  • b54cdc9a0f break_when_any_done==True for batch_size==1 Alexander Soare 2024-03-19 19:08:25 +0000
  • 46ac87d2a6 ready for review Alexander Soare 2024-03-19 18:59:08 +0000
  • 9946203071 Don't use batch dimension in normalization Alexander Soare 2024-03-19 18:52:36 +0000
  • 896a11f60e backup wip Alexander Soare 2024-03-19 18:50:04 +0000
  • 2d5abbbd6f
    Merge pull request #36 from Cadene/user/rcadene/2024_03_19_replay_buffer_folder Remi 2024-03-19 18:04:29 +0100
  • 7d5d99e036 Address more comments Cadene 2024-03-19 16:53:07 +0000
  • b420ab88f4 version naming conventions Cadene 2024-03-19 16:44:19 +0000
  • e799dc5e3f Improve mock_dataset Cadene 2024-03-19 16:38:07 +0000
  • 10034e85c4 Aloha done Cadene 2024-03-19 16:03:42 +0000
  • ea17f4ce50 backup wip Alexander Soare 2024-03-19 16:02:09 +0000
  • 6a1a29386a Add replay_buffer directory in pusht datasets + aloha (WIP) Cadene 2024-03-19 15:49:45 +0000
  • 9cdc24bc0e WIP user/rcadene/2024_03_14_video_dataset Cadene 2024-03-19 13:41:49 +0000
  • 88347965c2 revert dp changes, make act and tdmpc batch friendly Alexander Soare 2024-03-18 19:18:21 +0000
  • 09ddd9bf92 Merge branch 'main' into user/alexander-soare/multistep_policy_and_serial_env Alexander Soare 2024-03-18 18:27:50 +0000
  • a346469a5a Update time to monotonic Cadene 2024-03-18 16:26:07 +0000
  • 2bef00c317 Add video decoding in dataset (WIP: issue with gray background) Cadene 2024-03-18 16:25:33 +0000
  • 9954994a4b Add video decoding in dataset (WIP: issue with gray background) Cadene 2024-03-18 16:24:32 +0000
  • 0fc94b81b3 Add video decoding in dataset (WIP: issue with gray background) Cadene 2024-03-18 16:24:05 +0000
  • 099a465367
    Merge pull request #35 from Cadene/switch_between_train_and_eval Remi 2024-03-18 13:21:16 +0100
  • 8e346b379d switch between train and eval Alexander Soare 2024-03-18 09:45:17 +0000
  • d32a279435 Add test.py for gpu decoder Cadene 2024-03-13 16:27:44 +0000
  • bae7e7b41c Merge remote-tracking branch 'origin/main' into user/alexander-soare/multistep_policy_and_serial_env Alexander Soare 2024-03-15 14:06:53 +0000
  • 75cc10198f
    Merge pull request #31 from Cadene/disable_wandb_artifact Remi 2024-03-15 15:06:16 +0100
  • 3124f71ebd Merge remote-tracking branch 'origin/main' into user/alexander-soare/multistep_policy_and_serial_env Alexander Soare 2024-03-15 14:04:23 +0000
  • 4ecfd17f9e fix wandb artifact name and add disable option Alexander Soare 2024-03-15 13:56:55 +0000
  • 58d1787ee3
    Merge pull request #30 from Cadene/user/rcadene/2024_03_15_fix_path Remi 2024-03-15 14:41:33 +0100
  • b752833f3f fix download Cadene 2024-03-15 13:19:18 +0000
  • d374873849 use Path type instead of str fix_path Alexander Soare 2024-03-15 13:15:34 +0000
  • a45896dc8d Merge remote-tracking branch 'origin/main' into user/alexander-soare/multistep_policy_and_serial_env Alexander Soare 2024-03-15 13:05:35 +0000
  • 9c88071bc7
    Merge pull request #28 from Cadene/user/rcadene/2024_03_14_hf_dataset Remi 2024-03-15 13:52:13 +0100
  • 5805a7ffb1 small fix in type + comments Cadene 2024-03-15 12:44:52 +0000
  • 41521f7e96 self.root is Path or None + The following packages are already present in the pyproject.toml and will be skipped: Cadene 2024-03-15 10:56:46 +0000
  • b10c9507d4 Small fix Cadene 2024-03-15 00:36:55 +0000
  • a311d38796 Add aloha + improve readme Cadene 2024-03-15 00:30:11 +0000
  • 19730b3412 Add pusht on hf dataset (WIP) Cadene 2024-03-14 16:59:37 +0000
  • a222c88c99 Merge branch 'user/alexander-soare/train_pusht' into user/alexander-soare/multistep_policy_and_serial_env Alexander Soare 2024-03-14 16:06:21 +0000
  • 736bc969ca Merge branch 'main' into user/alexander-soare/train_pusht Alexander Soare 2024-03-14 16:05:11 +0000
  • 4822d63dbe Merge branch 'main' into user/alexander-soare/multistep_policy_and_serial_env Alexander Soare 2024-03-14 16:04:40 +0000
  • ba91976944 wip: still needs batch logic for act and tdmp Alexander Soare 2024-03-14 15:22:55 +0000
  • 95e84079ef
    Merge pull request #25 from Cadene/user/aliberts/2024_03_13_ci_fix Simon Alibert 2024-03-14 15:24:56 +0100
  • 8e856f1bf7 Update readme Simon Alibert 2024-03-14 15:24:38 +0100
  • 8c2b47752a Remove cuda env copy Simon Alibert 2024-03-14 13:55:35 +0100
  • f515cb6efd Add dm-control Simon Alibert 2024-03-14 13:42:03 +0100
  • c3f8d14fd8 CI env fix Simon Alibert 2024-03-14 13:27:22 +0100
  • 315cbcb422 wip - requesting feedback user/alexander/multistep_policy_rollout Alexander Soare 2024-03-13 12:54:31 +0000
  • 98484ac68e ready for review Alexander Soare 2024-03-12 21:59:01 +0000
  • 9512d1d2f3 Merge branch 'main' into user/alexander-soare/train_pusht Alexander Soare 2024-03-12 19:41:27 +0000
  • 8c56770318
    Merge pull request #11 from Cadene/user/rcadene/2024_03_06_aloha_env Remi 2024-03-12 16:02:48 +0100
  • 998dd2b874 info Cadene 2024-03-12 14:45:35 +0000
  • 7331df81d2 info Cadene 2024-03-12 14:44:23 +0000
  • 2c5d49cad5 Add sudo apt-get install -y libegl1-mesa-dev Cadene 2024-03-12 14:30:28 +0000
  • 5881eec376 Add DEVICE constant from LEROBOT_TESTS_DEVICE Cadene 2024-03-12 14:14:39 +0000
  • 29c73844b1 Update test.yml Cadene 2024-03-12 13:52:53 +0000
  • f9258898ff CI fix attempt Simon Alibert 2024-03-12 12:24:10 +0100
  • 9d002032d1 Add Aloha env and ACT policy Remi Cadene 2024-03-08 09:47:39 +0000
  • 060bac7672
    Merge pull request #23 from Cadene/kashif-patch-1 Remi 2024-03-12 10:11:42 +0100
  • 337208f28d
    Update README.md Kashif Rasul 2024-03-12 10:05:05 +0100
  • 87fcc536f9 wip - still need to verify full training run Alexander Soare 2024-03-11 18:45:21 +0000
  • 48e70e044e
    Merge pull request #18 from Cadene/user/aliberts/2024_03_11_dynamic_build_versioning Simon Alibert 2024-03-11 19:06:51 +0100
  • 4449c06823
    Add comment Simon Alibert 2024-03-11 19:06:22 +0100
  • 304355c917 Merge remote-tracking branch 'origin/main' into train_pusht Alexander Soare 2024-03-11 15:37:37 +0000
  • 706453ac17 WIP user/rcadene/2024_03_11_aloha_load_pretrained_from_act_repo Cadene 2024-03-11 14:06:34 +0000
  • bdd2c801bc PushtEnv inheriates AbstractEnv, Improve factory Normalization Cadene 2024-03-11 14:05:23 +0000
  • 2a01487494 early training loss as expected Alexander Soare 2024-03-11 13:34:04 +0000
  • a94800fc8a
    Merge pull request #20 from Cadene/user/aliberts/2024_03_11_license Simon Alibert 2024-03-11 14:32:05 +0100
  • a207b416b7 Typo fix Simon Alibert 2024-03-11 14:20:05 +0100
  • 78690d197f
    Merge pull request #19 from Cadene/user/aliberts/2024_03_11_wandb_config Simon Alibert 2024-03-11 14:17:44 +0100
  • 6d6c84b4a3
    Remove entity from config Simon Alibert 2024-03-11 14:14:17 +0100
  • 772a826bf2 Add FOWM & ALOHA copyright notices Simon Alibert 2024-03-11 13:54:20 +0100
  • 2cb8ae5037 Add acknowledgements Simon Alibert 2024-03-11 13:49:08 +0100
  • ebd5c786f1 fix import Cadene 2024-03-11 12:41:14 +0000
  • 2a4bc7371e fix training Cadene 2024-03-11 12:33:15 +0000
  • fab2b3240b
    Merge pull request #17 from Cadene/user/rcadene/2024_03_11_bugfix_compute_stats Remi 2024-03-11 13:44:07 +0100
  • 84a1647c01 fix import Cadene 2024-03-11 12:41:14 +0000
  • ccd5dc5a42 fix training Cadene 2024-03-11 12:33:15 +0000
  • c1e9c13ade Add license Simon Alibert 2024-03-11 13:31:52 +0100
  • 00fe4f4f18 Configure wandb entity outside config Simon Alibert 2024-03-11 13:09:46 +0100
  • 225eebde40 Add dynamic build versioning Simon Alibert 2024-03-11 12:10:46 +0100
  • 816b2e9d63 fix more bugs in normalization Cadene 2024-03-11 11:03:13 +0000
  • ac5571d609 fix more bugs in normalization Cadene 2024-03-11 11:03:13 +0000
  • a7ef4a6a33 fix bug in compute_stats for action normalization Cadene 2024-03-11 09:37:56 +0000
  • 570cb3af30 fix bug in compute_stats for action normalization Cadene 2024-03-11 09:37:56 +0000
  • d5336886cd poetry lock Cadene 2024-03-10 22:08:11 +0000
  • 66323dcd0a Add Aloha env and ACT policy Remi Cadene 2024-03-08 09:47:39 +0000
  • 7bf36cd413 Add AbstractEnv, Refactor AlohaEnv, Add rendering_hook in env, Minor modifications, (TODO: Refactor Pusht and Simxarm) user/rcadene/2024_03_08_act_policy Cadene 2024-03-10 22:00:48 +0000
  • d4ea4f0ad1
    Merge pull request #16 from Cadene/user/aliberts/2024_03_09_integrate_diffusion_policy Remi 2024-03-10 17:02:16 +0100
  • f54ee7cda0 Fix paths Simon Alibert 2024-03-10 16:51:50 +0100
  • 134009f337 Remove init files Simon Alibert 2024-03-10 16:38:49 +0100
  • 7982425670 Remove diffusion-policy dependency Simon Alibert 2024-03-10 16:36:30 +0100
  • 6c867d78ef Integrate pusht env from diffusion Simon Alibert 2024-03-10 16:33:03 +0100
  • 302b78962c Integrate diffusion policy Simon Alibert 2024-03-10 15:31:17 +0100
  • b49f7b70e2 Add tasks without end_effector that are compatible with dataset, Eval can run (TODO: training and pretrained model) Cadene 2024-03-10 10:52:12 +0000
  • 59397fb44a Move tdmpc files Simon Alibert 2024-03-09 18:44:36 +0100
  • f1230cdac0 Training can runs (TODO: eval) Cadene 2024-03-09 16:52:08 +0000
  • 1cc621ec36
    Merge pull request #15 from Cadene/user/aliberts/2024_03_09_add_ci_eval_test Simon Alibert 2024-03-09 16:54:50 +0100