lerobot/lerobot/common/policies
Michel Aractingi d1cc9665da - Refactor observation encoder in `modeling_sac.py`
- added `torch.compile` to the actor and learner servers.
- organized imports in `train_sac.py`
- optimized the parameters push by not sending the frozen pre-trained encoder.

Co-authored-by: Adil Zouitine <adilzouitinegm@gmail.com>
2025-03-24 13:44:55 +01:00
..
act User/pepijn/2025 03 17 act different image shapes (#870) 2025-03-18 11:09:05 +01:00
diffusion Add typos checks (#770) 2025-02-25 23:51:15 +01:00
hilserl Extend reward classifier for multiple camera views (#626) 2025-03-24 13:41:10 +01:00
pi0 refactor(config): Move device & amp args to PreTrainedConfig (#812) 2025-03-06 17:59:28 +01:00
sac - Refactor observation encoder in `modeling_sac.py` 2025-03-24 13:44:55 +01:00
tdmpc Add typos checks (#770) 2025-02-25 23:51:15 +01:00
vqbet Add typos checks (#770) 2025-02-25 23:51:15 +01:00
__init__.py chore(doc): add license header to all files (#818) 2025-03-05 17:56:51 +01:00
factory.py Change SAC policy implementation with configuration and modeling classes 2025-03-24 13:42:06 +01:00
normalize.py LeRobotDataset v2.1 (#711) 2025-02-25 15:27:29 +01:00
pretrained.py refactor(config): Move device & amp args to PreTrainedConfig (#812) 2025-03-06 17:59:28 +01:00
utils.py Simplify configs (#550) 2025-01-31 13:57:37 +01:00