Commit Graph

  • 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
  • a7ef4a6a33 fix bug in compute_stats for action normalization Cadene 2024-03-11 09:37:56 +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
  • 59397fb44a Move tdmpc files Simon Alibert 2024-03-09 18:44:36 +01: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
  • 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
  • e990f3e148 Merge pull request #6 from Cadene/user/rcadene/2024_03_04_diffusion Remi 2024-03-04 18:30:40 +01: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
  • d3aae3111c Merge pull request #5 from Cadene/user/aliberts/cpu_run Simon Alibert 2024-03-03 13:06:39 +01:00
  • b859e89936 Fix for PR #5 Simon Alibert 2024-03-03 13:05:21 +01:00
  • b33ec5a630 Add run on cpu-only compatibility Simon Alibert 2024-03-03 12:47:26 +01:00
  • 48ded3dbc7 fix Remi Cadene 2024-03-02 18:11:50 +00:00
  • 661bda45ea imagenet_norm: False Remi Cadene 2024-03-02 17:18:58 +00:00
  • 80785f8d0e Small fix, Refactor diffusion, Diffusion runs (TODO: remove normalization in diffusion) Remi Cadene 2024-03-02 17:04:39 +00:00
  • 45b4ecb727 pre-commit run -a Remi Cadene 2024-03-02 15:58:21 +00:00
  • 1ae6205269 Add Normalize, non_blocking=True in tdmpc, tdmpc run (TODO: diffusion) Remi Cadene 2024-03-02 15:53:29 +00:00
  • b5a2f460ea fix bus error Remi Cadene 2024-03-01 14:20:55 +00:00
  • fa7f473142 Merge pull request #4 from Cadene/user/aliberts/pusht_buffer_auto_download Simon Alibert 2024-03-01 15:20:30 +01:00
  • c1942d45d3 Fixes for PR #4 Simon Alibert 2024-03-01 14:59:05 +01:00
  • b862145e22 Added pusht dataset auto-download Simon Alibert 2024-03-01 14:31:54 +01:00
  • ca948c1e5b fix zip strict=False Cadene 2024-03-01 00:45:23 +00:00
  • ae050d2e94 Solve conflicts + pre-commit run -a Cadene 2024-02-29 23:31:32 +00:00
  • 0b9027f05e Clean logging, Refactor Cadene 2024-02-29 23:13:06 +00:00
  • cb7b375526 Merge pull request #3 from Cadene/pre-commit Simon Alibert 2024-02-29 21:48:19 +01:00
  • 2c05b75f45 Fixes for PR #3 Simon Alibert 2024-02-29 21:46:41 +01:00
  • 7e024fdce6 Ran pre-commit run --all-files Simon Alibert 2024-02-29 13:37:48 +01:00
  • 0e0e760e43 Added pre-commit Simon Alibert 2024-02-29 12:26:35 +01:00
  • afb6b8679e Merge pull request #2 from Cadene/install Remi 2024-02-28 19:47:03 +01:00
  • 5798979e9c comment Cadene 2024-02-28 18:31:54 +00:00
  • d5bfeadcf1 Added poetry setup Simon Alibert 2024-02-28 19:11:29 +01:00
  • ac90b9c3ee Fix diffusion (rm transpose), Add prefetch Cadene 2024-02-28 17:45:01 +00:00
  • cf5063e50e Add diffusion policy (train and eval works, TODO: reproduce results) Cadene 2024-02-28 15:21:30 +00:00
  • f1708c8a37 install fix Simon Alibert 2024-02-28 12:35:49 +01:00
  • 98f8869743 WIP Simon Alibert 2024-02-28 10:57:08 +01:00
  • e543c9a42c small fix % Cadene 2024-02-27 11:54:31 +00:00
  • 7df542445c Small fix and improve logging message Cadene 2024-02-27 11:44:26 +00:00
  • 21670dce90 Refactor train, eval_policy, logger, Add diffusion.yaml (WIP) Cadene 2024-02-26 01:10:09 +00:00
  • 5a219fed6e Refactor policy config Cadene 2024-02-25 18:26:44 +00:00
  • b16c334825 Refactor configs to have env in seperate yaml + Fix training Cadene 2024-02-25 17:42:47 +00:00
  • eec134d72b small change Cadene 2024-02-25 14:20:14 +00:00
  • ed80db2846 Sanitize cfg.env Cadene 2024-02-25 12:02:29 +00:00
  • 9b469c4232 Fix pusht.yaml Cadene 2024-02-25 11:49:39 +00:00
  • 0eb9b5d1a5 Sanitize cfg.wandb Cadene 2024-02-25 11:15:09 +00:00
  • 5812298290 fix style readme Cadene 2024-02-25 11:09:47 +00:00