Commit Graph

17 Commits

Author SHA1 Message Date
Remi Cadene d82c59dfc3 tests/data/save_policy_to_safetensors/xarm_tdmpc/actions.safetensors: convert to Git LFS 2024-05-19 19:58:46 +00:00
Remi Cadene 8eb8642384 tests/data/save_policy_to_safetensors/pusht_diffusion/param_stats.safetensors: convert to Git LFS 2024-05-19 19:58:40 +00:00
Remi Cadene 1e2d6099f2 tests/data/save_policy_to_safetensors/pusht_diffusion/output_dict.safetensors: convert to Git LFS 2024-05-19 19:58:33 +00:00
Remi Cadene 290931d872 tests/data/save_policy_to_safetensors/pusht_diffusion/grad_stats.safetensors: convert to Git LFS 2024-05-19 19:58:27 +00:00
Remi Cadene 931c5f9168 tests/data/save_policy_to_safetensors/pusht_diffusion/actions.safetensors: convert to Git LFS 2024-05-19 19:57:42 +00:00
Remi Cadene a5098d7bf3 update aloha 2024-05-19 19:43:21 +00:00
Remi Cadene 155b3fdcf2 update aloha_act 2024-05-19 19:30:20 +00:00
Remi Cadene 62ec7227e1 revert tests/data/save_policy_to_safetensors/ 2024-05-19 17:53:42 +00:00
Remi Cadene 1d9433b411 WIP
Add hf-transfer, num_workers 8

gc collect, rmtree save_to_disk False

Track .safetensors in git lfs

run save_dataset_to_safetensors on all datasets

Update datasets

Update download_raw
2024-05-17 02:11:28 +00:00
Alexander Soare 4d7d41cdee
Fix act action queue (#185) 2024-05-16 15:43:25 +01:00
Akshay Kashyap c9069df9f1
Port SpatialSoftmax and remove Robomimic dependency (#182)
Co-authored-by: Alexander Soare <alexander.soare159@gmail.com>
2024-05-16 15:34:10 +01:00
Alexander Soare 1249aee3ac
Enable logging all the information returned by the `forward` methods of policies (#151) 2024-05-10 07:45:32 +01:00
Alexander Soare 473345fdf6
Fix stats override in ACT config (#161) 2024-05-09 15:16:47 +01:00
Alexander Soare e89521dfa0
Enable tests for TD-MPC (#160) 2024-05-09 13:42:12 +01:00
Alexander Soare a8e245fb31
Remove loss masking from diffusion policy (#135) 2024-05-06 07:27:01 +01:00
Alexander Soare f3bba0270d
Remove EMA model from Diffusion Policy (#134) 2024-05-05 11:26:12 +01:00
Simon Alibert c77633c38c
Add regression tests (#119)
- Add `tests/scripts/save_policy_to_safetensor.py` to generate test artifacts
- Add `test_backward_compatibility to test generated outputs from the policies against artifacts
2024-05-04 16:20:30 +02:00