Commit Graph

8 Commits

Author SHA1 Message Date
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
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
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