Skip to content
Snippets Groups Projects
Unverified Commit 942ddb69 authored by Yinggang Wang's avatar Yinggang Wang Committed by GitHub
Browse files

Export Optimizer and _LRScheduler (#5156)


* feat(*): export Optimizer and _LRScheduler

* fix(*): import module

* fix

Co-authored-by: default avataroneflow-ci-bot <69100618+oneflow-ci-bot@users.noreply.github.com>
parent 09e19644
No related branches found
No related tags found
No related merge requests found
......@@ -14,9 +14,12 @@ See the License for the specific language governing permissions and
limitations under the License.
"""
from oneflow.python.oneflow_export import oneflow_export, experimental_api
from .optimizer import Optimizer
@oneflow_export("optim.lr_scheduler._LRScheduler")
@experimental_api
class LrScheduler(object):
def __init__(self, optimizer, last_step=-1, verbose=False):
if not isinstance(optimizer, Optimizer):
......
......@@ -18,6 +18,7 @@ import warnings
from typing import Dict, Callable, Union, Any, Iterator
from types import GeneratorType
from oneflow.python.oneflow_export import oneflow_export, experimental_api
from oneflow.python.nn.parameter import Parameter
from oneflow.python.framework.tensor import Tensor
......@@ -54,6 +55,8 @@ class ParamGroup(object):
return self._parameters
@oneflow_export("optim.Optimizer")
@experimental_api
class Optimizer(object):
def __init__(self):
self.param_groups = list()
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment