keras/keras_core/optimizers/optimizer.py

21 lines
552 B
Python
Raw Normal View History

2023-04-09 19:21:45 +00:00
from keras_core import backend
from keras_core.api_export import keras_core_export
from keras_core.optimizers import base_optimizer
if backend.backend() == "tensorflow":
from keras_core.backend.tensorflow import optimizer as tf_optimizer
2023-05-13 01:59:21 +00:00
2023-05-17 23:06:01 +00:00
BackendOptimizer = tf_optimizer.TFOptimizer
else:
2023-05-26 16:13:08 +00:00
BackendOptimizer = base_optimizer.BaseOptimizer
2023-05-17 23:06:01 +00:00
keras_core_export(["keras_core.Optimizer", "keras_core.optimizers.Optimizer"])
class Optimizer(BackendOptimizer):
pass
base_optimizer_keyword_args = base_optimizer.base_optimizer_keyword_args