-
-
Notifications
You must be signed in to change notification settings - Fork 617
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
auto_optim enhancement #2169
auto_optim enhancement #2169
Conversation
@Chandan-h-509 Thank you ! It would be nice to have a test of this new feature. |
@sdesrozis I have added the test case. |
@Chandan-h-509 a minor comment to address and a format error to fix. See https://github.com/pytorch/ignite/pull/2169/checks?check_run_id=3389112372 Otherwise, it looks good 👍🏻 |
@sdesrozis I have made the final changes. |
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.
@Chandan-h-509 Thank you ! LGTM !
cc @sandylaker for visibility |
@vfdev-5 I have made the new change as requested. |
Fixes #2168
Description: Add kwargs to the auto_optim argument and pass it to DistributedOptimizer .
Check list: