lerobot/lerobot/common/policies
Alexander Soare f3bba0270d
Remove EMA model from Diffusion Policy (#134)
2024-05-05 11:26:12 +01:00
..
act Remove batch length restrictions in select_action (#123) 2024-05-04 15:33:55 +02:00
diffusion Remove EMA model from Diffusion Policy (#134) 2024-05-05 11:26:12 +01:00
tdmpc Add regression tests (#119) 2024-05-04 16:20:30 +02:00
factory.py Fix missing `policy.to(device)` in policy factory (#126) 2024-05-01 17:26:58 +01:00
normalize.py Use PytorchModelHubMixin to save models as safetensors (#125) 2024-05-01 16:17:18 +01:00
policy_protocol.py Refactor TD-MPC (#103) 2024-05-01 16:40:04 +01:00
utils.py ready for review 2024-04-12 11:36:52 +01:00