add-custom-act-policy#126
Open
wilsonchenghy wants to merge 1 commit into
Open
Conversation
Contributor
There was a problem hiding this comment.
Pull request overview
Adds support for running a locally trained “custom ACT” policy checkpoint in the SO101 vial-task Isaac Lab eval script, and introduces a new humanoid_act Python package (training, dataset utilities, checkpointing, and a vendored ACT/DETR backbone) under autonomy/il.
Changes:
- Extend
lerobot_eval.pywith acustom_actpolicy type that loads a local.ptcheckpoint and runs inference in sim. - Add
humanoid_actpackage with training (humanoid-act-train), smoke test (humanoid-act-smoke), checkpoint I/O, normalization, and sim stepping integration. - Update
autonomy/ilpackaging to shiphumanoid_actand expose new CLI entry points.
Reviewed changes
Copilot reviewed 20 out of 20 changed files in this pull request and generated 6 comments.
Show a summary per file
| File | Description |
|---|---|
| autonomy/simulation/so101_vial_task/scripts/lerobot_eval.py | Adds custom_act option and wires sim eval to a local ACT policy. |
| autonomy/il/pyproject.toml | Exposes new humanoid-act-* scripts and packages humanoid_act. |
| autonomy/il/humanoid_act/train.py | Implements a basic training loop for ACT on LeRobot datasets. |
| autonomy/il/humanoid_act/smoke.py | Adds a quick dataloader smoke-test entry point. |
| autonomy/il/humanoid_act/policy.py | Wraps ACT model + loss (L1 + KL) and image normalization. |
| autonomy/il/humanoid_act/normalize.py | Loads/saves normalization stats and reads them from LeRobot meta. |
| autonomy/il/humanoid_act/eval.py | Implements LocalCustomACTPolicy used by lerobot_eval.py. |
| autonomy/il/humanoid_act/detr/util/misc.py | Vendored DETR/torchvision utilities (distributed helpers, etc.). |
| autonomy/il/humanoid_act/detr/util/init.py | Initializes vendored util package. |
| autonomy/il/humanoid_act/detr/models/transformer.py | Vendored transformer implementation. |
| autonomy/il/humanoid_act/detr/models/position_encoding.py | Vendored positional encoding implementation. |
| autonomy/il/humanoid_act/detr/models/detr_vae.py | Implements DETR-VAE model used by ACT. |
| autonomy/il/humanoid_act/detr/models/backbone.py | Vendored backbone wrapper around torchvision ResNet. |
| autonomy/il/humanoid_act/detr/models/init.py | Exposes model builders for the vendored code. |
| autonomy/il/humanoid_act/detr/build.py | Builds ACT model + optimizer without argparse side effects. |
| autonomy/il/humanoid_act/detr/init.py | Top-level entry for the vendored DETR/ACT module. |
| autonomy/il/humanoid_act/dataset.py | Converts LeRobot datasets into ACT training batches. |
| autonomy/il/humanoid_act/config.py | Defines ACTConfig and derives it from dataset metadata. |
| autonomy/il/humanoid_act/checkpoint.py | Save/load checkpoint payloads, config, and stats. |
| autonomy/il/humanoid_act/init.py | Re-exports humanoid_act public API symbols. |
💡 Add Copilot custom instructions for smarter, more guided reviews. Learn how to get started.
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
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.
No description provided.