lunny/keras
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix CI

Browse Source
This commit is contained in:
Francois Chollet 2023-08-18 22:34:04 -07:00
parent 59321fc09a
commit 35f5cf8659
2 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
keras_core/backend/tensorflow/trainer.py
View File

@ -63,7 +63,7 @@ class TensorFlowTrainer(base_trainer.Trainer):
# Compute gradients
if self.trainable_weights:
trainable_weights = [v.value for v in self.trainable_weights]
trainable_weights = self.trainable_weights
gradients = tape.gradient(loss, trainable_weights)
# Update weights

6
keras_core/optimizers/base_optimizer.py
View File

@ -175,11 +175,15 @@ class BaseOptimizer:
"""
initializer = initializers.Zeros()
name = name or "var"
if hasattr(reference_variable, "path"):
name = reference_variable.path.replace("/", "_") + "_" + name
else:
name = reference_variable.name + "_" + name
return self.add_variable(
shape=reference_variable.shape,
initializer=initializer,
dtype=reference_variable.dtype,
name=reference_variable.path.replace("/", "_") + "_" + name,
name=name,
)
def _check_variables_are_known(self, variables):