lerobot/lerobot/common/policies
Alexander Soare d36ca387e8 backup wip 2024-05-08 11:31:54 +01:00
..
act Remove batch length restrictions in select_action (#123) 2024-05-04 15:33:55 +02:00
diffusion Remove loss masking from diffusion policy (#135) 2024-05-06 07:27:01 +01:00
tdmpc backup wip 2024-05-08 11:31:54 +01: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