Skip to content

Implement validation_step to call training_step#36

Open
mtiezzi wants to merge 1 commit into
nvidia-cosmos:mainfrom
mtiezzi:main
Open

Implement validation_step to call training_step#36
mtiezzi wants to merge 1 commit into
nvidia-cosmos:mainfrom
mtiezzi:main

Conversation

@mtiezzi

@mtiezzi mtiezzi commented Nov 5, 2025

Copy link
Copy Markdown

This fix allows the model inhering from it (e.g. Video2WorldModelRectifiedFlow) to perform the validation step. Inspired by Predict2

@mharrim-nv

Copy link
Copy Markdown
Contributor

@arslananvidia can you review this change or re-assign.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants