File size: 870 Bytes
1f5470c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 |
from keras.src import backend
from keras.src.api_export import keras_export
from keras.src.optimizers import base_optimizer
if backend.backend() == "tensorflow":
from keras.src.backend.tensorflow.optimizer import (
TFOptimizer as BackendOptimizer,
)
elif backend.backend() == "torch":
from keras.src.backend.torch.optimizers import (
TorchOptimizer as BackendOptimizer,
)
elif backend.backend() == "jax":
from keras.src.backend.jax.optimizer import JaxOptimizer as BackendOptimizer
else:
class BackendOptimizer(base_optimizer.BaseOptimizer):
pass
@keras_export(["keras.Optimizer", "keras.optimizers.Optimizer"])
class Optimizer(BackendOptimizer, base_optimizer.BaseOptimizer):
pass
Optimizer.__doc__ = base_optimizer.BaseOptimizer.__doc__
base_optimizer_keyword_args = base_optimizer.base_optimizer_keyword_args
|