Commit Graph

  • 0a33a414fb grad_clip_norm as arg of update policy qgallouedec/move_update_optim_schedul_outside_policy Quentin Gallouédec 2024-04-25 12:12:06 +02:00
  • cd76980d50 fix update Quentin Gallouédec 2024-04-25 12:05:33 +02:00
  • e760e4cd63 Move normalization to policy for act and diffusion (#90) Remi 2024-04-25 11:47:38 +02:00
  • 1ffc0e0d94 format Quentin Gallouédec 2024-04-25 11:46:17 +02:00
  • fda092d5bf DS_Store in gitignore Quentin Gallouédec 2024-04-25 11:42:30 +02:00
  • 2a9ea01d5a merge update functions Quentin Gallouédec 2024-04-25 11:42:20 +02:00
  • 0bca982fca move optimizer and scheduler outside policies Quentin Gallouédec 2024-04-25 11:26:40 +02:00
  • dae901f556 move update outside policies Quentin Gallouédec 2024-04-25 10:45:10 +02:00
  • 7626b9a4a3 Refactor datasets into LeRobotDataset Cadene 2024-04-21 12:00:32 +00:00
  • 6d56bcb5de improve docstring Cadene 2024-04-24 21:40:42 +00:00
  • 0ec28bf71a address comments Cadene 2024-04-24 20:57:09 +00:00
  • bc96284ca0 remove last update Cadene 2024-04-24 16:48:55 +00:00
  • 343579368b nit Cadene 2024-04-24 15:56:22 +00:00
  • 72751b7cf6 make load_state_dict work Cadene 2024-04-24 15:40:09 +00:00
  • 0660f71556 make unit tests pass Cadene 2024-04-23 21:39:39 +00:00
  • 42ed7bb670 Move normalize/unnormalize transforms to policy for act and diffusion Cadene 2024-04-20 21:08:14 +00:00
  • c1bcf857c5 Fix online training (#94) Remi 2024-04-23 18:54:55 +02:00
  • 1030ea0070 Loads episode_data_index and stats during dataset __init__ (#85) Remi 2024-04-23 14:13:25 +02:00
  • 7cee7a0f20 Add mobile Aloha and visu with rerun.io user/rcadene/2024_04_20_mobile_aloha_rerun Remi Cadene 2024-04-20 16:19:55 +02:00
  • 2a59825a00 fix online training Cadene 2024-04-20 00:12:34 +00:00
  • 06628ba059 fix online training Cadene 2024-04-19 23:58:38 +00:00
  • b2b5329683 fix online training Cadene 2024-04-19 23:48:43 +00:00
  • 85f1554da8 fix visualize_dataset Cadene 2024-04-19 23:40:35 +00:00
  • 9b4c2e2a9f small fix Cadene 2024-04-19 23:30:39 +00:00
  • 20928021c0 Add tests/data Cadene 2024-04-19 23:19:16 +00:00
  • c20cf2fbbc Remove Prod, Tests are passind Cadene 2024-04-19 23:18:45 +00:00
  • 35a573c98e Use v1.1, hf_transform_to_torch, Add 3 xarm datasets Cadene 2024-04-19 18:17:13 +00:00
  • 714a776277 id -> index, finish moving compute_stats before hf_dataset push_to_hub Cadene 2024-04-19 10:33:42 +00:00
  • 64b09ea7a7 WIP add load functions + episode_data_index Cadene 2024-04-18 23:54:52 +00:00
  • 0bd2ca8d82 Add meta_data, revision v1.1 Cadene 2024-04-18 22:18:57 +00:00
  • e2168163cd Quality of life patches for eval.py (#86) Alexander Soare 2024-04-19 12:33:47 +01:00
  • ac0ab27333 Hotfix test_examples.py (#87) Simon Alibert 2024-04-19 12:36:04 +02:00
  • 8d980940a2 Fix tolerance for delta_timestamps (#84) Alexander Soare 2024-04-18 18:48:22 +01:00
  • 7ad1909641 Tests cleaning & simplification (#81) Simon Alibert 2024-04-18 14:47:42 +02:00
  • 0928afd37d Improve dataset examples (#82) Remi 2024-04-18 11:43:16 +02:00
  • d5c4b0c344 Merge pull request #80 from huggingface/alexander-soare/unify_policy_api Alexander Soare 2024-04-17 17:05:22 +01:00
  • dd9c6eed15 Add temporary patch in TD-MPC Alexander Soare 2024-04-17 16:27:57 +01:00
  • 2298ddf226 wip Alexander Soare 2024-04-17 16:21:37 +01:00
  • 63e5ec6483 revert some formatting changes Alexander Soare 2024-04-17 11:01:01 +01:00
  • c50a13ab31 draft Alexander Soare 2024-04-17 10:50:54 +01:00
  • 18dd8f32cd Merge remote-tracking branch 'upstream/main' into unify_policy_api Alexander Soare 2024-04-17 09:10:30 +01:00
  • 6d78bca749 Merge pull request #77 from huggingface/alexander-soare/fix_save_stats Alexander Soare 2024-04-17 09:09:18 +01:00
  • 296bbfe1ad Merge branch 'fix_stats_saving' into unify_policy_api Alexander Soare 2024-04-17 09:08:04 +01:00
  • c9454333d8 revision Alexander Soare 2024-04-17 09:02:35 +01:00
  • 1331d3b4e4 fix issue with saving freshly computed stats Alexander Soare 2024-04-17 08:49:28 +01:00
  • bff4b673c9 Merge remote-tracking branch 'upstream/main' into unify_policy_api Alexander Soare 2024-04-17 08:08:57 +01:00
  • 3f1c322d56 Merge pull request #73 from huggingface/user/rcadene/2024_04_14_hf_datasets Remi 2024-04-16 21:54:37 +02:00
  • fbc31d906c Merge pull request #74 from huggingface/user/aliberts/2024_04_15_setup_contributions Simon Alibert 2024-04-16 19:51:13 +02:00
  • 4057cc6c28 Apply suggestions from code review Simon Alibert 2024-04-16 19:35:01 +02:00
  • 91badebdfc fix tests Cadene 2024-04-16 17:29:31 +00:00
  • 4327e43f19 fix merge thingy Cadene 2024-04-16 17:24:25 +00:00
  • 36d9e885ef Address comments Cadene 2024-04-16 17:14:40 +00:00
  • b241ea46dd move download_and_upload_dataset.py to root_dir Cadene 2024-04-16 16:14:35 +00:00
  • e09d25267e fix online training Cadene 2024-04-16 16:07:39 +00:00
  • 4a3eac4743 fix unit tests, stats was missing, visualize_dataset was broken Cadene 2024-04-16 12:53:31 +00:00
  • 69eeced9d9 add datasets to poetry/cpu Cadene 2024-04-16 12:29:58 +00:00
  • 0980fff6cc HF datasets works Cadene 2024-04-16 12:20:38 +00:00
  • 5edd9a89a0 Move stats_dataset init into else statement -> faster init Cadene 2024-04-16 07:04:08 +00:00
  • c7a8218620 typo Cadene 2024-04-15 21:28:34 +00:00
  • 67d79732f9 Add download_and_upload_dataset.py in script, update all datasets, update online training Cadene 2024-04-15 21:26:33 +00:00
  • c6aca7fe44 For Pusht: use hf datasets to train, rename load_data_with_delta_timestamps -> load_previous_and_future_frames Cadene 2024-04-15 10:08:10 +00:00
  • cb3978b5f3 backup wip Alexander Soare 2024-04-16 18:12:39 +01:00
  • 4ed55c3ba3 Merge pull request #76 from huggingface/user/aliberts/2024_04_16_refactor_pyproject Remi 2024-04-16 18:55:09 +02:00
  • 0eb899de73 Merge remote-tracking branch 'upstream/main' into unify_policy_api Alexander Soare 2024-04-16 17:30:41 +01:00
  • a8ddefcfd4 Merge pull request #71 from huggingface/user/alexander-soare/refactor_dp Alexander Soare 2024-04-16 17:24:05 +01:00
  • a9496fde39 revision 1 Alexander Soare 2024-04-16 17:15:51 +01:00
  • 8a322da422 backup wip Alexander Soare 2024-04-16 16:35:04 +01:00
  • 23be5e1e7b backup wip Alexander Soare 2024-04-16 16:31:44 +01:00
  • 43a614c173 Fix test_examples Alexander Soare 2024-04-16 14:07:16 +01:00
  • 9c2f10bd04 ready for review Alexander Soare 2024-04-16 13:43:58 +01:00
  • 03b08eb74e backup wip Alexander Soare 2024-04-16 12:51:32 +01:00
  • 38ef878eed Add COC badge Simon Alibert 2024-04-16 13:37:20 +02:00
  • dee174f678 pre-commit autoupdate Simon Alibert 2024-04-16 12:10:26 +02:00
  • 70e8de95f7 Clean & update pyproject Simon Alibert 2024-04-16 12:09:56 +02:00
  • 376d75f8d3 Add env info Simon Alibert 2024-04-16 10:35:43 +02:00
  • a621ec8d88 Add PR & issue templates Simon Alibert 2024-04-16 10:34:29 +02:00
  • f9b48abe64 Add CONTRIBUTING Simon Alibert 2024-04-16 10:33:58 +02:00
  • 352010b718 Add COC Simon Alibert 2024-04-16 10:32:09 +02:00
  • 5608e659e6 backup wip Alexander Soare 2024-04-15 19:06:44 +01:00
  • 14f3ffb412 Merge remote-tracking branch 'upstream/main' into refactor_dp Alexander Soare 2024-04-15 17:08:28 +01:00
  • e37f4e8c53 Merge pull request #72 from huggingface/alexander-soare/policy_config Alexander Soare 2024-04-15 16:44:21 +01:00
  • 30023535f9 revision 1 Alexander Soare 2024-04-15 10:56:43 +01:00
  • 40d417ef60 Make sure to make remove all traces of omegaconf from policy config Alexander Soare 2024-04-15 09:59:18 +01:00
  • 9241b5e830 pass step as kwarg Alexander Soare 2024-04-15 09:52:54 +01:00
  • 2ccf89d78c try fix tests Alexander Soare 2024-04-15 09:47:25 +01:00
  • ef4bd9e25c Use dataclass config for ACT Alexander Soare 2024-04-15 09:39:23 +01:00
  • 34f00753eb remove policy.py Alexander Soare 2024-04-12 17:13:25 +01:00
  • 55e484124a draft pr Alexander Soare 2024-04-12 16:55:32 +01:00
  • 6d0a45a97d ready for review Alexander Soare 2024-04-12 11:36:52 +01:00
  • 5666ec3ec7 backup wip Alexander Soare 2024-04-11 18:33:54 +01:00
  • 94cc22da9e Merge remote-tracking branch 'upstream/main' into refactor_dp Alexander Soare 2024-04-11 17:52:10 +01:00
  • 976a197f98 backup wip Alexander Soare 2024-04-11 17:51:35 +01:00
  • 5bd953e8e7 Merge pull request #64 from huggingface/user/rcadene/2024_03_31_remove_torchrl Remi 2024-04-11 16:45:16 +02:00
  • c1a618e567 fix pusht images type from float32 to uint8, update gym-pusht dependencies Cadene 2024-04-11 14:29:16 +00:00
  • 4216636084 fix Cadene 2024-04-11 14:01:23 +00:00
  • 8e5b4365ac fix Cadene 2024-04-11 13:57:22 +00:00
  • 84d2468da1 fix Cadene 2024-04-11 13:51:13 +00:00
  • 5be83fbff6 fix Cadene 2024-04-11 13:46:13 +00:00
  • a605eec7e9 Use pusht as example so it's fast Cadene 2024-04-11 13:36:50 +00:00
  • 2f4af32d3f small fix Cadene 2024-04-11 13:21:06 +00:00