Simon Alibert
|
4631d36c05
|
Add get_safe_torch_device in policies
|
2024-03-20 18:38:55 +01:00 |
|
Alexander Soare
|
896a11f60e
|
backup wip
|
2024-03-19 18:50:04 +00:00 |
|
Alexander Soare
|
88347965c2
|
revert dp changes, make act and tdmpc batch friendly
|
2024-03-18 19:18:21 +00:00 |
|
Alexander Soare
|
a222c88c99
|
Merge branch 'user/alexander-soare/train_pusht' into user/alexander-soare/multistep_policy_and_serial_env
|
2024-03-14 16:06:21 +00:00 |
|
Alexander Soare
|
ba91976944
|
wip: still needs batch logic for act and tdmp
|
2024-03-14 15:24:10 +00:00 |
|
Alexander Soare
|
2a01487494
|
early training loss as expected
|
2024-03-11 13:34:04 +00:00 |
|
Simon Alibert
|
302b78962c
|
Integrate diffusion policy
|
2024-03-10 15:31:17 +01:00 |
|
Simon Alibert
|
a6d353c419
|
Fix
|
2024-03-05 17:00:17 +01:00 |
|
Remi Cadene
|
cfc304e870
|
Refactor env queue, Training diffusion works (Still not converging)
|
2024-03-04 11:00:51 +00:00 |
|
Remi Cadene
|
fddd9f0311
|
Add possibility for the policy to provide a sequence of actions to the env
|
2024-03-03 14:02:24 +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
|
80785f8d0e
|
Small fix, Refactor diffusion, Diffusion runs (TODO: remove normalization in diffusion)
|
2024-03-02 17:04:39 +00:00 |
|