Merge pull request #16731 from miker2241:patch-add-variable-definitions

PiperOrigin-RevId: 458500460
keras-team · Jul 1, 2022 · 7dd01b1 · 7dd01b1
2 parents cc28c7d + 5591008
commit 7dd01b1
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/keras/optimizers/optimizer_v2/optimizer_v2.py b/keras/optimizers/optimizer_v2/optimizer_v2.py
@@ -115,6 +115,8 @@ class OptimizerV2(tf.__internal__.tracking.Trackable):
     opt = tf.keras.optimizers.SGD(learning_rate=0.1)
     # `loss` is a callable that takes no argument and returns the value
     # to minimize.
+    var1 = tf.Variable(2.0)
+    var2 = tf.Variable(5.0)
     loss = lambda: 3 * var1 * var1 + 2 * var2 * var2
     # In graph mode, returns op that minimizes the loss by updating the listed
     # variables.