Simon Alibert
|
6c867d78ef
|
Integrate pusht env from diffusion
|
2024-03-10 16:33:03 +01:00 |
|
Remi Cadene
|
a027f4edfb
|
Add cfg.offline_prioritized_sampler
|
2024-03-04 23:08:52 +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
|
0f2fa4d9ef
|
Add obs queue to pusht, Set n_obs_steps=2 for diffusion (Not fully tested)
|
2024-03-03 13:21:31 +00:00 |
|
Remi Cadene
|
661bda45ea
|
imagenet_norm: False
|
2024-03-02 17:18:58 +00:00 |
|
Cadene
|
0b9027f05e
|
Clean logging, Refactor
|
2024-02-29 23:21:27 +00:00 |
|
Cadene
|
ac90b9c3ee
|
Fix diffusion (rm transpose), Add prefetch
|
2024-02-28 17:45:01 +00:00 |
|
Cadene
|
cf5063e50e
|
Add diffusion policy (train and eval works, TODO: reproduce results)
|
2024-02-28 15:21:42 +00:00 |
|
Cadene
|
7df542445c
|
Small fix and improve logging message
|
2024-02-27 11:44:26 +00:00 |
|
Cadene
|
21670dce90
|
Refactor train, eval_policy, logger, Add diffusion.yaml (WIP)
|
2024-02-26 01:10:09 +00:00 |
|