hangX/lerobot
1
0
Fork 0
forked from tangger/lerobot
Code Pull Requests Activity

Add maniskill support.

Browse Source 
Co-authored-by: Michel Aractingi <michel.aractingi@gmail.com>
This commit is contained in:
AdilZouitine
2025-02-14 19:53:29 +00:00
parent 291358d6a2
commit b7a0ffc3b8
6 changed files with 222 additions and 27 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
lerobot/scripts/server/learner_server.py
View File

@@ -362,7 +362,7 @@ def add_actor_information_and_train(
# If cfg.resume, shift the interaction step with the last checkpointed step in order to not break the logging
interaction_message["Interaction step"] += interaction_step_shift
logger.log_dict(interaction_message, mode="train", custom_step_key="Interaction step")
logging.info(f"Interaction message: {interaction_message}")
# logging.info(f"Interaction message: {interaction_message}")
if len(replay_buffer) < cfg.training.online_step_before_learning:
continue