This commit is contained in:
Nick committed 2019-10-29 15:23:19 -06:00
1 parent 35cdb4ae52
commit 3969b73cec
1 file changed
+1 -1
+1 -1
View File
@@ -142,7 +142,7 @@ def loss(labels, logits):
# the optimizer is not used since this code only supports inference
# however, to compile the model, we still define it
optimizer = tf.contrib.estimator.clip_gradients_by_norm(
tf.train.AdagradOptimizer(learning_rate=1e-2), 0.25)
tf.train.AdagradOptimizer(learning_rate=1e-4), 0.25)
# compile the model with the optimizer and loss
model.compile(optimizer=optimizer, loss=loss)