Add unified policy evaluation for RL checkpoints - #522
Draft
acrlw wants to merge 10 commits into
Draft
Conversation
- Add --original-task to select the recorded EmbodiChain task. - Keep task metrics in the final evaluation result. - Close task resources when evaluator setup or execution fails. - Write the run manifest after training completes successfully.
- Let explicit config paths override run manifest entries. - Store implicit recorder output under each run videos directory. - Build task policy context before applying Viewer reset behavior. - Document config precedence and the generated output layout.
Add one eval-policy command for Headless, Viewer, and external Profile workflows. Reuse the training runtime, record run manifests, and consolidate the focused documentation and tests.
Use hybrid, fast-rt, and offline-rt consistently across simulation configuration, CLIs, policy evaluation, documentation, and focused tests.
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.
Description
This PR adds
embodichain eval-policyas the post-training entry point for EmbodiChain checkpoints and external Policy Profiles.The change:
run-manifest.json;--viewer, using the same observation flattening, deterministic Policy inference, action conversion, reset, and termination behavior as Headless evaluation;evaluations/<timestamp>-policy/evaluation.json; and.ptexample, Viewer controls, documentation, and focused tests.flowchart LR Run[Training run] --> Manifest[run-manifest.json] Manifest --> Runtime[EmbodiChain RL runtime] Runtime --> Headless[Headless evaluation] Runtime --> Bridge[Task Viewer bridge] Profile[External Motion Profile] --> Evaluator[MotionPolicyEvaluator] Bridge --> Evaluator Headless --> Report[evaluation.json] Evaluator --> ReportEvaluate an EmbodiChain training run:
Run the external Policy example:
Viewer controls use
Backspaceto reset the task and camera,Tto switch between tracking and free camera modes,Rto start or stop recording, andEscto close the Viewer. The ANYmal-C example uses W/S forvx, A/D forvy, Q/E foryaw, and M to clear the command.No linked issue.
Type of change
Screenshots
Validation
main.evaluation.json.evaluation.json..ptcompleted the selected control steps and wroteevaluation.json.git diff --checkpassed.Checklist
black .command to format the code base.