KEMBAR78
Fix the dropout onnx symbolic, and ensure all exported models in test_operators.py are eval mode by houseroad · Pull Request #16547 · pytorch/pytorch · GitHub
Skip to content

Conversation

@houseroad
Copy link
Member

In eval mode, skip dropout operator in onnx exporter.

Copy link
Contributor

@facebook-github-bot facebook-github-bot left a comment

Choose a reason for hiding this comment

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

@dzhulgakov has imported this pull request. If you are a Facebook employee, you can view this diff on Phabricator.

@dzhulgakov
Copy link
Collaborator

Good catch. Looks like the bug was there from the very beginning - even before #10384 refactoring

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.

4 participants