2023-04-09 19:21:45 +00:00
|
|
|
from keras_core import backend
|
2023-05-12 22:44:03 +00:00
|
|
|
from keras_core.optimizers import base_optimizer
|
2023-04-12 18:31:58 +00:00
|
|
|
from keras_core.api_export import keras_core_export
|
2023-05-01 02:51:01 +00:00
|
|
|
|
|
|
|
|
|
2023-05-12 22:44:03 +00:00
|
|
|
if backend.backend() == 'tensorflow':
|
|
|
|
|
from keras_core.backend.tensorflow import optimizer as tf_optimizer
|
|
|
|
|
Optimizer = tf_optimizer.TFOptimizer
|
|
|
|
|
else:
|
|
|
|
|
Optimizer = base_optimizer.Optimizer
|
2023-05-01 02:51:01 +00:00
|
|
|
|
|
|
|
|
|
2023-05-12 22:44:03 +00:00
|
|
|
keras_core_export([
|
|
|
|
|
"keras_core.Optimizer", "keras_core.optimizers.Optimizer"])(Optimizer)
|
2023-05-01 02:51:01 +00:00
|
|
|
|
|
|
|
|
|
2023-05-12 22:44:03 +00:00
|
|
|
base_optimizer_keyword_args = base_optimizer.base_optimizer_keyword_args
|
