Skip to content

Conversation

@azrael417
Copy link
Collaborator

restricting inputs to predefined MLP models and adding sanity checks and meaningful error messages.

  • Added CriticMLP which expects exactly 2 inputs (state, actor). Using this in the tests
  • Making more thorough sanity checks for the RL models

Copy link
Collaborator

@romerojosh romerojosh left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Just a couple of suggested comment edits, but otherwise looks good to me. We will need to follow up with another PR to document these (and the base MLP model) once I make the other MLP changes.

@romerojosh
Copy link
Collaborator

/build_and_test

@github-actions
Copy link

🚀 Build workflow triggered! View run

@github-actions
Copy link

✅ Build workflow passed! View run

@romerojosh romerojosh merged commit 87129b4 into NVIDIA:master Sep 18, 2025
3 checks passed
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