Validation fu added to examples/structural_mechanics/crash/train.py

[dakhare-creator]

PhysicsNeMo Pull Request

Description

Checklist

• I am familiar with the Contributing Guidelines.
• New or existing tests cover these changes.
• The documentation is up to date with these changes.
• The CHANGELOG.md is up to date with these changes.
• An issue is linked to this pull request.

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.

[greptile-apps]

Greptile Overview

Greptile Summary

This PR adds validation functionality to the structural mechanics crash simulation training example. The main changes include: (1) adding validation dataset creation and distributed sampling in train.py, (2) implementing a validation loop that computes time-step-wise MSE loss and aggregates results across distributed ranks, (3) adding validation configuration parameters to control validation frequency and checkpoint saving, and (4) refactoring the inference code to use a unified sample object interface instead of passing individual graph components separately.

The validation implementation follows distributed training best practices by properly handling data sampling, metric aggregation, and logging only on rank 0. The changes integrate cleanly with the existing training pipeline and tensorboard logging infrastructure, providing essential model monitoring capabilities for the crash simulation example.

PR Description Notes:

• The PR description is largely empty with only unchecked checklist items
• No standalone description of changes provided
• No linked issues or changelog updates mentioned
• Missing information about new dependencies or testing coverage

Important Files Changed

Filename	Score	Overview
examples/structural_mechanics/crash/train.py	4/5	Added comprehensive validation functionality with distributed sampling, MSE computation, and tensorboard logging
examples/structural_mechanics/crash/conf/training/default.yaml	5/5	Added validation configuration parameters for sample count, validation frequency, and checkpoint saving
examples/structural_mechanics/crash/inference.py	4/5	Refactored model forward pass to use unified sample object interface instead of separate graph components

_{3 files reviewed, 3 comments}

_{Edit Code Review Agent Settings | Greptile}

[mnabian]

/blossom-ci

[mnabian]

/blossom-ci

[mnabian]

LGTM! Thanks for addressing the comments!