Change LR in examples (#19)

keras-team · Jan 30, 2025 · 676378f · 676378f
1 parent ab0e362
commit 676378f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/examples/basic_retrieval.py b/examples/basic_retrieval.py
@@ -296,7 +296,7 @@ def compute_loss(self, x, y, y_pred, sample_weight, training=True):
 """
 
 model = RetrievalModel(users_count + 1, movies_count + 1)
-model.compile(optimizer=keras.optimizers.Adagrad(learning_rate=1.0))
+model.compile(optimizer=keras.optimizers.Adagrad(learning_rate=0.1))
 
 """
 Then train the model. Evaluation takes a bit of time, so we only evaluate the

diff --git a/examples/data_parallel_retrieval.py b/examples/data_parallel_retrieval.py
@@ -188,7 +188,7 @@ def compute_loss(self, x, y, y_pred, sample_weight, training=True):
 """
 
 model = RetrievalModel(users_count + 1, movies_count + 1)
-model.compile(optimizer=keras.optimizers.Adagrad(learning_rate=1.0))
+model.compile(optimizer=keras.optimizers.Adagrad(learning_rate=0.1))
 
 """
 Let's train the model. Evaluation takes a bit of time, so we only evaluate the