Policy Normalization Migration (PR #1452)
Breaking Change: LeRobot policies no longer have built-in normalization layers embedded in their weights. Normalization is now handled by externalPolicyProcessorPipeline components.
What Changed?
| Before PR #1452 | After PR #1452 | |
|---|---|---|
| Normalization Location | Embedded in model weights (normalize_inputs.*) | External PolicyProcessorPipeline components |
| Model State Dict | Contains normalization statistics | Clean weights only - no normalization parameters |
| Usage | policy(batch) handles everything | preprocessor(batch) → policy(...) → postprocessor(...) |
Impact on Existing Models
- Models trained before PR #1452 have normalization embedded in their weights
- These models need migration to work with the new
PolicyProcessorPipelinesystem - The migration extracts normalization statistics and creates separate processor pipelines
Migrating Old Models
Use the migration script to convert models with embedded normalization:- Extracts normalization statistics from model weights
- Creates external preprocessor and postprocessor pipelines
- Removes normalization layers from model weights
- Saves clean model + processor pipelines
- Pushes to Hub with automatic PR creation
Using Migrated Models
Hardware API Redesign (PR #777)
PR #777 improves the LeRobot calibration but is not backward-compatible. Below is an overview of what changed and how you can continue to work with datasets created before this pull request.What Changed?
| Before PR #777 | After PR #777 | |
|---|---|---|
| Joint range | Degrees -180...180° | Normalised range Joints: –100...100 Gripper: 0...100 |
| Zero position (SO100 / SO101) | Arm fully extended horizontally | In middle of the range for each joint |
| Boundary handling | Software safeguards to detect ±180 ° wrap-arounds | No wrap-around logic needed due to mid-range zero |
Impact on Existing Datasets
- Recorded trajectories created before PR #777 will replay incorrectly if loaded directly:
- Joint angles are offset and incorrectly normalized.
- Any models directly finetuned or trained on the old data will need their inputs and outputs converted.
Using Datasets Made with Previous Calibration
We provide a migration example script for replaying an episode recorded with the previous calibration here:examples/backward_compatibility/replay.py.
Below we take you through the modifications that are done in the example script to make the previous calibration datasets work.
1. Update Key Names
New codebase uses.pos suffix for the position observations and we have removed main_ prefix:
2. Adjust Shoulder Lift
For"shoulder_lift" (id = 2), the 0 position is changed by -90 degrees and the direction is reversed compared to old calibration/code.
3. Adjust Elbow Flex
For"elbow_flex" (id = 3), the 0 position is changed by -90 degrees compared to old calibration/code.
4. Enable Degrees Normalization
To use degrees normalization we then set the--robot.use_degrees option to true.
Using Policies Trained with Previous Calibration
Policies output actions in the same format as the datasets (torch.Tensors). Therefore, the same transformations should be applied.
To find these transformations, we recommend to first try and replay an episode of the dataset your policy was trained on using the section above.
Then, add these same transformations on your inference script (shown here in the record.py script):
Migration Checklist
Identify Breaking Changes
Check which PR affects your models/datasets:
- Models trained before PR #1452: Need policy normalization migration
- Datasets recorded before PR #777: Need hardware calibration migration
Test
Verify the migrated model/dataset works correctly:
- For datasets: Replay a sample episode
- For models: Run inference on test data
Getting Help
Discord Support
Ask migration questions in the LeRobot Discord server
GitHub Issues
Report migration bugs or compatibility issues
Version Compatibility Matrix
| LeRobot Version | Policy Normalization | Hardware Calibration | Python Version |
|---|---|---|---|
| v2.0+ | External Pipeline | New (±100 range) | ≥3.12 |
| v1.5 - v1.9 | Embedded | New (±100 range) | ≥3.10 |
| v1.0 - v1.4 | Embedded | Old (±180 range) | ≥3.10 |
Always check the GitHub releases for detailed changelog information.