Commit Graph

5 Commits

Author SHA1 Message Date
Akshay Kashyap c0a7b9cfb9 update only pusht test save tensors 2024-05-13 15:58:49 -07:00
Akshay Kashyap 1bfddb3619 save new policy safetensors 2024-05-13 14:37:57 -07: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