-
Notifications
You must be signed in to change notification settings - Fork 129
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Changed TheseusLayer.forward() to receive optimizer_kwargs as a single dict #45
Conversation
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Looks good.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Can you add a test to show that a user can no longer pass in aux_vars
as a keyword argument to forward? I am not sure I see where that is explicitly prevented with the current restructuring..
Added. In the current version nothing should be accepted besides |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM!
…e dict (facebookresearch#45) * [refactor] Changed TheseusLayer so that optimizer_kwargs are passed as a single dict. * Updated all tutorials to use optimizer_kwargs dict in forward(). * Updated examples to use optimizer_kwargs dict in forward(). * Add additional test to check that TheseusLayer.forward(aux_vars=) is not accepted.
Motivation and Context
Closes #42
Closes #28
How Has This Been Tested
All tests passes and also added new test to check that kwargs are passed correctly. Also updated tutorials and examples.
Types of changes
Checklist