Simon Alibert
|
89eaab140b
|
Add pusht test artifact
|
2024-03-09 15:36:20 +01:00 |
|
Simon Alibert
|
f1e2837d63
|
fix pusht data_dir path
|
2024-03-08 12:26:15 +01:00 |
|
Remi Cadene
|
f95ecd66fc
|
Improve visualize_dataset, Improve AbstractReplayBuffer, Small improvements
|
2024-03-06 10:15:57 +00:00 |
|
Remi Cadene
|
2f80d71c3e
|
Remove noqa-F821
|
2024-03-05 10:22:21 +00:00 |
|
Remi Cadene
|
d4e0849970
|
Refactor datasets with abstract class
|
2024-03-05 10:20:57 +00:00 |
|
Remi Cadene
|
e29fbb50e8
|
Fix grad_clip_norm 0 -> 10, Fix normalization min_max to be per channel
|
2024-03-04 17:26:34 +00:00 |
|
Remi Cadene
|
cfc304e870
|
Refactor env queue, Training diffusion works (Still not converging)
|
2024-03-04 11:00:51 +00:00 |
|
Remi Cadene
|
cbbed590a9
|
Add mode to NormalizeTransform with mean_std or min_max (Not fully tested)
|
2024-03-03 13:19:02 +00:00 |
|
Remi Cadene
|
45b4ecb727
|
pre-commit run -a
|
2024-03-02 15:58:21 +00:00 |
|
Remi Cadene
|
1ae6205269
|
Add Normalize, non_blocking=True in tdmpc, tdmpc run (TODO: diffusion)
|
2024-03-02 15:53:29 +00:00 |
|
Remi Cadene
|
b5a2f460ea
|
fix bus error
|
2024-03-01 14:22:05 +00:00 |
|
Simon Alibert
|
c1942d45d3
|
Fixes for PR #4
|
2024-03-01 14:59:05 +01:00 |
|
Simon Alibert
|
b862145e22
|
Added pusht dataset auto-download
|
2024-03-01 14:31:54 +01:00 |
|
Cadene
|
ae050d2e94
|
Solve conflicts + pre-commit run -a
|
2024-02-29 23:31:32 +00:00 |
|
Cadene
|
0b9027f05e
|
Clean logging, Refactor
|
2024-02-29 23:21:27 +00:00 |
|
Simon Alibert
|
7e024fdce6
|
Ran pre-commit run --all-files
|
2024-02-29 13:37:48 +01:00 |
|
Cadene
|
591985c67d
|
Fix done in pusht, Fix --time in sbatch
|
2024-02-22 17:58:26 +00:00 |
|
Cadene
|
ece89730e6
|
Add pusht dataset (TODO verify reward is aligned), Refactor visualize_dataset, Add video_dir, fps, state_dim, action_dim to config (Training works)
|
2024-02-21 00:49:40 +00:00 |
|
Cadene
|
3da6ffb2cb
|
Fix unit tests, Refactor, Add pusht env, (TODO pusht replay buffer, image preprocessing)
|
2024-02-20 12:26:57 +00:00 |
|