lerobot/lerobot/common/policies
Michel Aractingi 98c473a41e Added support for checkpointing the policy. We can save and load the policy state dict, optimizers state, optimization step and interaction step
Added functions for converting the replay buffer from and to LeRobotDataset. When we want to save the replay buffer, we convert it first to LeRobotDataset format and save it locally and vice-versa.

Co-authored-by: Adil Zouitine <adilzouitinegm@gmail.com>
2025-03-24 13:44:10 +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 Added support for checkpointing the policy. We can save and load the policy state dict, optimizers state, optimization step and interaction step 2025-03-24 13:44:10 +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