Skip to content
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

[onert/python] Introduce optimizer API #14543

Open
wants to merge 2 commits into
base: master
Choose a base branch
from

Conversation

ragmani
Copy link
Contributor

@ragmani ragmani commented Jan 13, 2025

This commit introduces optimizer API.

  • adam.py : Adam optimizer class
  • sgd.py : SGB optimizer class

ONE-DCO-1.0-Signed-off-by: ragmani [email protected]

This commit introduces optimizer API.
  - adam.py : Adam optimizer class
  - sgd.py  : SGB optimizer class

ONE-DCO-1.0-Signed-off-by: ragmani <[email protected]>
@ragmani ragmani requested a review from a team January 13, 2025 06:57
@ragmani
Copy link
Contributor Author

ragmani commented Jan 13, 2025

For #14505
Draft : #14492

@ragmani ragmani added the PR/ready for review It is ready to review. Please review it. label Jan 13, 2025
Comment on lines +5 to +7
"""
Base class for optimizers. Subclasses should implement the `step` method.
"""
Copy link
Contributor

@zetwhite zetwhite Jan 15, 2025

Choose a reason for hiding this comment

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

Suggested change
"""
Base class for optimizers. Subclasses should implement the `step` method.
"""
"""
Base class for optimizers.
"""

Since step() is deleted, this needs to be updated.

Copy link
Contributor

Choose a reason for hiding this comment

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

@ragmani But I think this has to be fixed. That's why I didn't approve this PR yet.

Copy link
Contributor

@hseok-oh hseok-oh left a comment

Choose a reason for hiding this comment

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

LGTM

Copy link
Contributor

@zetwhite zetwhite left a comment

Choose a reason for hiding this comment

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

LGTM

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
PR/ready for review It is ready to review. Please review it.
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants