lerobot/lerobot/common/policies/sac
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
..
configuration_sac.py - Added additional logging information in wandb around the timings of the policy loop and optimization loop. 2025-03-24 13:43:28 +01:00
modeling_sac.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 13:44:10 +01:00