Unified Aero Recipe: Fix for manifest-based path without augmentation#1755
Merged
coreyjadams merged 25 commits intoJun 29, 2026
Merged
Conversation
…ither `inverse()` or `inverse_td()`. And never used
…b.com/peterdsharpe/physicsnemo into psharpe/unified-recipe-minor-cleanups
- Removed the `_all_reduce` method from `MetricCalculator` as it was redundant. - Updated the `__call__` method to directly return the metrics without reduction. - Introduced `_reduce_and_average_epoch` function in `train.py` to handle averaging of epoch loss and metrics across distributed processes. - Added unit tests for the new `_reduce_and_average_epoch` function to ensure correctness in both single-process and distributed scenarios.
- Added `_build_manifest_val_dataset` function to create a separate validation dataset when augmentations are enabled, ensuring validation does not inherit training augmentations. - Updated `build_dataloaders` to utilize the new validation dataset logic, allowing for consistent evaluation behavior across manifest and directory modes. - Introduced unit tests for `_build_manifest_val_dataset` to verify correct behavior for both augmented and non-augmented scenarios.
melo-gonzo
approved these changes
Jun 25, 2026
melo-gonzo
left a comment
Collaborator
There was a problem hiding this comment.
I had a look over the changes in dataset.py - all LGTM! Thanks @peterdsharpe
…tion. (Previously, was scattered across infer.py and train.py)
…-recipe-manifest-validation-fix
Contributor
Greptile SummaryThis PR fixes a bug in manifest mode where validation data was being passed through the augmented transform chain when
Important Files Changed
Reviews (2): Last reviewed commit: "Merge branch 'main' into psharpe/unified..." | Re-trigger Greptile |
…ll_reduce logic is shared for iters and epochs.
…-recipe-manifest-validation-fix
Collaborator
Author
coreyjadams
approved these changes
Jun 29, 2026
coreyjadams
left a comment
Collaborator
There was a problem hiding this comment.
Looks reasonable to me!
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.
PhysicsNeMo Pull Request
Description
Stack on top of #1753, #1754
Checklist
Dependencies
Review Process
All PRs are reviewed by the PhysicsNeMo team before merging.
Depending on which files are changed, GitHub may automatically assign a maintainer for review.
We are also testing AI-based code review tools (e.g., Greptile), which may add automated comments with a confidence score.
This score reflects the AI’s assessment of merge readiness and is not a qualitative judgment of your work, nor is
it an indication that the PR will be accepted / rejected.
AI-generated feedback should be reviewed critically for usefulness.
You are not required to respond to every AI comment, but they are intended to help both authors and reviewers.
Please react to Greptile comments with 👍 or 👎 to provide feedback on their accuracy.