Commit Graph

  • 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
  • e765e26b0b Sanitize cfg.policy, Fix skip_frame pusht.yaml Cadene 2024-02-25 11:09:02 +00:00
  • fc4b98544b Add tests to Readme Cadene 2024-02-25 10:52:31 +00:00
  • 6f5c731936 Rename test -> tests Cadene 2024-02-25 10:51:07 +00:00
  • 598bb496b0 Add policies/factory, Add test, Add _self_ in config Cadene 2024-02-25 10:50:23 +00:00
  • 64b5920e94 format Cadene 2024-02-24 18:19:18 +00:00
  • aed02dc7c6 Add multithreading for video generation, Speed policy sampling Cadene 2024-02-24 18:18:39 +00:00
  • 591985c67d Fix done in pusht, Fix --time in sbatch Cadene 2024-02-22 17:58:26 +00:00
  • 664cfb2023 Add sbatch.sh Cadene 2024-02-22 13:04:32 +00:00
  • 63d18475cc fix simxarm factory Cadene 2024-02-22 13:04:24 +00:00
  • 96c53ad06f remove comments Cadene 2024-02-22 12:15:14 +00:00
  • e3643d6146 Wandb works, One output dir Cadene 2024-02-22 12:14:12 +00:00
  • ece89730e6 Add pusht dataset (TODO verify reward is aligned), Refactor visualize_dataset, Add video_dir, fps, state_dim, action_dim to config (Training works) Cadene 2024-02-21 00:49:40 +00:00
  • 3dc14b5576 Add Prod transform, Add test_factory Cadene 2024-02-20 14:22:16 +00:00
  • 3da6ffb2cb Fix unit tests, Refactor, Add pusht env, (TODO pusht replay buffer, image preprocessing) Cadene 2024-02-20 12:26:57 +00:00
  • fdfb2010fd black Cadene 2024-02-18 01:24:19 +00:00
  • a5c305a7a4 offline training + online finetuning converge to 33 reward! Cadene 2024-02-18 01:23:44 +00:00
  • 0b4084f0f8 Clean + alpha beta corresponds to config (before 0.7 and 0.9) Cadene 2024-02-16 16:27:54 +00:00
  • 0cdd23dcac Update README Cadene 2024-02-16 15:14:59 +00:00
  • c202c2b3c2 Online finetuning runs (sometimes crash because of nans) Cadene 2024-02-16 15:13:24 +00:00
  • 228c045674 Eval reproduced! Train running (but not reproduced) Cadene 2024-02-10 15:46:24 +00:00
  • 937b2f8cba Add option for random policy Cadene 2024-01-31 13:54:32 +00:00
  • 5a5b190f70 Add common, refactor eval with eval_policy Cadene 2024-01-31 13:48:12 +00:00
  • 1e52499490 eval.mp4 works! Cadene 2024-01-30 23:30:14 +00:00
  • 1144819c29 First real commit, simxarm env added with torchrl! Cadene 2024-01-29 12:49:30 +00:00
  • 0396980450 .gitignore Cadene 2024-01-29 12:49:06 +00:00
  • 007ffa898f first commit Cadene 2024-01-26 15:51:11 +00:00