21 lines
552 B
Python
21 lines
552 B
Python
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
|
|
|
|
BackendOptimizer = tf_optimizer.TFOptimizer
|
|
else:
|
|
BackendOptimizer = base_optimizer.BaseOptimizer
|
|
|
|
|
|
keras_core_export(["keras_core.Optimizer", "keras_core.optimizers.Optimizer"])
|
|
|
|
|
|
class Optimizer(BackendOptimizer):
|
|
pass
|
|
|
|
|
|
base_optimizer_keyword_args = base_optimizer.base_optimizer_keyword_args
|