HIL-SERL tutorial for simulation fixed and expanded #2
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
What this does
This PR fixes two critical bugs in the HIL-SERL simulation training framework and adds comprehensive documentation:
Bug Fixes:
control_time_sparameter not being respected during training - episodes were always 10 seconds regardless of configurationrandom_block_positionflag not being properly passed to the gym environment, preventing cube randomizationImprovements:
hil_serl_simulation_training_guide_README.md) for HIL-SERL simulation trainingKey Changes:
TimeLimitWrapperclass to properly enforce episode time limits based oncontrol_time_sconfigurationrandom_block_positionparameter to environment configuration and properly passed it to gym environmentHow it was tested
"control_time_s": 40.0in configuration now properly limits episodes to 40 seconds instead of default 10 seconds"random_block_position": truenow properly randomizes cube positions between episodesTest Commands:
How to checkout & try? (for the reviewer)
# Modify control_time_s in hi_rl_test_gamepad.json to 20.0 and verify episodes last 20 seconds python -m lerobot.scripts.rl.gym_manipulator --config_path examples/hil_serl_simulation_training/hi_rl_test_gamepad.json# Check the new training guide cat examples/hil_serl_simulation_training/hil_serl_simulation_training_guide_README.mdExpected Behavior:
control_time_ssetting (30s in training config, 40s in recording config)random_block_position: true