Commit Graph

  • 060bac7672 Merge pull request #23 from Cadene/kashif-patch-1 Remi 2024-03-12 10:11:42 +01:00
  • 337208f28d Update README.md Kashif Rasul 2024-03-12 10:05:05 +01:00
  • 87fcc536f9 wip - still need to verify full training run Alexander Soare 2024-03-11 18:45:21 +00:00
  • 48e70e044e Merge pull request #18 from Cadene/user/aliberts/2024_03_11_dynamic_build_versioning Simon Alibert 2024-03-11 19:06:51 +01:00
  • 4449c06823 Add comment Simon Alibert 2024-03-11 19:06:22 +01:00
  • 304355c917 Merge remote-tracking branch 'origin/main' into train_pusht Alexander Soare 2024-03-11 15:37:37 +00:00
  • 706453ac17 WIP user/rcadene/2024_03_11_aloha_load_pretrained_from_act_repo Cadene 2024-03-11 14:06:34 +00:00
  • bdd2c801bc PushtEnv inheriates AbstractEnv, Improve factory Normalization Cadene 2024-03-11 14:05:23 +00:00
  • 2a01487494 early training loss as expected Alexander Soare 2024-03-11 13:34:04 +00:00
  • a94800fc8a Merge pull request #20 from Cadene/user/aliberts/2024_03_11_license Simon Alibert 2024-03-11 14:32:05 +01:00
  • a207b416b7 Typo fix Simon Alibert 2024-03-11 14:20:05 +01:00
  • 78690d197f Merge pull request #19 from Cadene/user/aliberts/2024_03_11_wandb_config Simon Alibert 2024-03-11 14:17:44 +01:00
  • 6d6c84b4a3 Remove entity from config Simon Alibert 2024-03-11 14:14:17 +01:00
  • 772a826bf2 Add FOWM & ALOHA copyright notices Simon Alibert 2024-03-11 13:54:20 +01:00
  • 2cb8ae5037 Add acknowledgements Simon Alibert 2024-03-11 13:49:08 +01:00
  • ebd5c786f1 fix import Cadene 2024-03-11 12:41:14 +00:00
  • 2a4bc7371e fix training Cadene 2024-03-11 12:33:15 +00:00
  • fab2b3240b Merge pull request #17 from Cadene/user/rcadene/2024_03_11_bugfix_compute_stats Remi 2024-03-11 13:44:07 +01:00
  • 84a1647c01 fix import Cadene 2024-03-11 12:41:14 +00:00
  • ccd5dc5a42 fix training Cadene 2024-03-11 12:33:15 +00:00
  • c1e9c13ade Add license Simon Alibert 2024-03-11 13:31:52 +01:00
  • 00fe4f4f18 Configure wandb entity outside config Simon Alibert 2024-03-11 13:09:46 +01:00
  • 225eebde40 Add dynamic build versioning Simon Alibert 2024-03-11 12:10:46 +01:00
  • 816b2e9d63 fix more bugs in normalization Cadene 2024-03-11 11:03:13 +00:00
  • ac5571d609 fix more bugs in normalization Cadene 2024-03-11 11:03:13 +00:00
  • a7ef4a6a33 fix bug in compute_stats for action normalization Cadene 2024-03-11 09:37:56 +00:00
  • 570cb3af30 fix bug in compute_stats for action normalization Cadene 2024-03-11 09:37:56 +00:00
  • d5336886cd poetry lock Cadene 2024-03-10 22:08:11 +00:00
  • 66323dcd0a Add Aloha env and ACT policy Remi Cadene 2024-03-08 09:47:39 +00:00
  • 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 +00:00
  • d4ea4f0ad1 Merge pull request #16 from Cadene/user/aliberts/2024_03_09_integrate_diffusion_policy Remi 2024-03-10 17:02:16 +01:00
  • f54ee7cda0 Fix paths Simon Alibert 2024-03-10 16:51:50 +01:00
  • 134009f337 Remove init files Simon Alibert 2024-03-10 16:38:49 +01:00
  • 7982425670 Remove diffusion-policy dependency Simon Alibert 2024-03-10 16:36:30 +01:00
  • 6c867d78ef Integrate pusht env from diffusion Simon Alibert 2024-03-10 16:33:03 +01:00
  • 302b78962c Integrate diffusion policy Simon Alibert 2024-03-10 15:31:17 +01:00
  • 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 +00:00
  • 59397fb44a Move tdmpc files Simon Alibert 2024-03-09 18:44:36 +01:00
  • f1230cdac0 Training can runs (TODO: eval) Cadene 2024-03-09 16:52:08 +00:00
  • 1cc621ec36 Merge pull request #15 from Cadene/user/aliberts/2024_03_09_add_ci_eval_test Simon Alibert 2024-03-09 16:54:50 +01:00
  • 471ebfef62 Remove cleanup + reduce eval steps Simon Alibert 2024-03-09 16:50:17 +01:00
  • 30753d879c Fix Simon Alibert 2024-03-09 16:41:31 +01:00
  • c6fb40fb29 Add end-to-end pusht eval test Simon Alibert 2024-03-09 16:37:34 +01:00
  • fa7a947acc Merge pull request #12 from Cadene/user/aliberts/2024_03_08_test_data Simon Alibert 2024-03-09 16:00:39 +01:00
  • 450e32e4b5 update readme Simon Alibert 2024-03-09 15:57:29 +01:00
  • 0da85b2cef Removed debugging lines Simon Alibert 2024-03-09 15:51:42 +01:00
  • f2c7ab5b3b test ci Simon Alibert 2024-03-09 15:41:23 +01:00
  • cde866dac0 reupload Cadene 2024-03-09 14:18:47 +00:00
  • a54a0feb63 remove mock Cadene 2024-03-09 14:18:25 +00:00
  • f440a681ad Add mock_dataset.py Cadene 2024-03-09 14:13:31 +00:00
  • 35bd577deb Add mock_dataset.py Cadene 2024-03-09 14:13:14 +00:00
  • 327f60e4be Update readme Simon Alibert 2024-03-08 16:48:33 +01:00
  • 74ad9d5154 Change pusht artifacts path Simon Alibert 2024-03-08 16:48:07 +01:00
  • 89eaab140b Add pusht test artifact Simon Alibert 2024-03-08 15:54:30 +01:00
  • 7dbdbb051c Add git lfs in CI checkout Simon Alibert 2024-03-09 15:35:20 +01:00
  • 5395829596 Add act yaml (TODO: try train.py) Cadene 2024-03-08 18:08:28 +00:00
  • a45802c281 Remove download.py add a WIP for Simxarm Cadene 2024-03-08 18:07:49 +00:00
  • 167a51cb69 Remove download.py add a WIP for Simxarm Cadene 2024-03-08 18:07:33 +00:00
  • fbc66a082b Copy past from act repo Cadene 2024-03-08 16:54:43 +00:00
  • 603455e313 Update README Cadene 2024-03-08 16:15:56 +00:00
  • 6500945be5 Rendering works (fps look fast tho? TODO action bounding is too wide [-1,1]) Cadene 2024-03-08 15:33:35 +00:00
  • ebbcad8c05 WIP Aloha env tests pass Cadene 2024-03-08 14:37:23 +00:00
  • d98b435b4c WIP Remi Cadene 2024-03-08 09:47:39 +00:00
  • 4cc7e1539e Merge pull request #10 from Cadene/user/rcadene/2024_03_06_fix_tests Simon Alibert 2024-03-08 12:30:08 +01:00
  • f1e2837d63 fix pusht data_dir path Simon Alibert 2024-03-08 12:26:15 +01:00
  • 54b05bfb77 Comment aloha, add test to CI Simon Alibert 2024-03-07 15:57:38 +01:00
  • 524d29aa80 fix tests Remi Cadene 2024-03-06 13:55:12 +00:00
  • c2c0ef9927 Leaner caching strategy Simon Alibert 2024-03-07 13:00:47 +01:00
  • 66373e9b13 Merge pull request #9 from Cadene/user/rcadene/2024_03_05_aloha_dataset Remi 2024-03-06 14:51:55 +01:00
  • 7d33b437fa Add h5py Simon Alibert 2024-03-06 12:28:20 +01:00
  • b9dc3be463 Trigger CI on direct pushes Simon Alibert 2024-03-06 12:13:23 +01:00
  • 86ec62f98a Trigger CI on PR labeled CI Simon Alibert 2024-03-06 12:02:01 +01:00
  • 52bdfc659e Trigger CI on PR sync and reopen Simon Alibert 2024-03-06 11:35:19 +01:00
  • d782b029e1 Add aloha dataset Remi Cadene 2024-03-06 10:15:11 +00:00
  • 49c0955f97 Merge pull request #7 from Cadene/user/rcadene/2024_03_05_abstract_replay_buffer Remi 2024-03-06 11:25:24 +01:00
  • eed24b083a small fix Remi Cadene 2024-03-06 10:21:22 +00:00
  • f95ecd66fc Improve visualize_dataset, Improve AbstractReplayBuffer, Small improvements Remi Cadene 2024-03-06 10:14:03 +00:00
  • d34c0a3c49 Update CI caches Simon Alibert 2024-03-05 17:19:39 +01:00
  • 11a5a7ca45 Merge pull request #8 from Cadene/user/aliberts/2024_03_05_ci Simon Alibert 2024-03-05 17:15:48 +01:00
  • a6d353c419 Fix Simon Alibert 2024-03-05 17:00:17 +01:00
  • d6556e6519 Fix Simon Alibert 2024-03-05 16:40:05 +01:00
  • 12af67066d Fix Simon Alibert 2024-03-05 16:26:31 +01:00
  • 7a20ef65f6 Add CI, Remove old install files Simon Alibert 2024-03-05 16:03:03 +01:00
  • 2f80d71c3e Remove noqa-F821 Remi Cadene 2024-03-05 10:22:21 +00:00
  • d4e0849970 Refactor datasets with abstract class Remi Cadene 2024-03-05 10:20:57 +00:00
  • e132a267aa offline_prioritized_sampler: true Remi Cadene 2024-03-04 23:17:59 +00:00
  • a027f4edfb Add cfg.offline_prioritized_sampler Remi Cadene 2024-03-04 23:08:52 +00:00
  • 570f8d01df Revert "WIP" Remi Cadene 2024-03-04 22:41:47 +00:00
  • 7938adcdfc Add sbatch_hopper.sh Remi Cadene 2024-03-04 22:41:31 +00:00
  • 20c08bb740 WIP Remi Cadene 2024-03-04 17:32:29 +00:00
  • 2bcf2631b9 minor comment Remi Cadene 2024-03-04 22:34:44 +00:00
  • b169f93f3a WIP user/rcadene/2024_03_04_sbatch_hopper Remi Cadene 2024-03-04 17:32:29 +00:00
  • e990f3e148 Merge pull request #6 from Cadene/user/rcadene/2024_03_04_diffusion Remi 2024-03-04 18:30:40 +01:00
  • 10fc36a572 WIP user/rcadene/2024_03_04_diffusion_pretrained_from_repo Remi Cadene 2024-03-04 17:28:11 +00:00
  • e29fbb50e8 Fix grad_clip_norm 0 -> 10, Fix normalization min_max to be per channel Remi Cadene 2024-03-04 17:26:34 +00:00
  • cfc304e870 Refactor env queue, Training diffusion works (Still not converging) Remi Cadene 2024-03-04 10:59:43 +00:00
  • fddd9f0311 Add possibility for the policy to provide a sequence of actions to the env Remi Cadene 2024-03-03 14:02:24 +00:00
  • 4c400b41a5 Improve log msg in train.py Remi Cadene 2024-03-03 13:22:09 +00:00
  • 0f2fa4d9ef Add obs queue to pusht, Set n_obs_steps=2 for diffusion (Not fully tested) Remi Cadene 2024-03-03 13:21:31 +00:00
  • cbbed590a9 Add mode to NormalizeTransform with mean_std or min_max (Not fully tested) Remi Cadene 2024-03-03 13:19:02 +00:00