lerobot/lerobot/common/datasets
Michel Aractingi c504d23f82 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 09:13:12 +00:00
..
push_dataset_to_hub Update pre-commits (#733) 2025-02-15 15:51:17 +01:00
v2 Simplify configs (#550) 2025-01-31 13:57:37 +01:00
card_template.md Dataset v2.0 (#461) 2024-11-29 19:04:00 +01:00
compute_stats.py Dataset v2.0 (#461) 2024-11-29 19:04:00 +01:00
factory.py Update pre-commits (#733) 2025-02-15 15:51:17 +01:00
image_writer.py Fix example 6 (#572) 2024-12-11 10:32:18 +01:00
lerobot_dataset.py Reward classifier and training (#528) 2025-03-24 09:04:12 +00:00
online_buffer.py Dataset v2.0 (#461) 2024-11-29 19:04:00 +01:00
sampler.py add EpisodeAwareSampler (#217) 2024-05-31 13:43:47 +01:00
transforms.py Simplify configs (#550) 2025-01-31 13:57:37 +01:00
utils.py 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 09:13:12 +00:00
video_utils.py Dataset v2.0 (#461) 2024-11-29 19:04:00 +01:00