From 3969b73cece04b20915258726f543415f868cbf7 Mon Sep 17 00:00:00 2001 From: Nick Date: Tue, 29 Oct 2019 15:23:19 -0600 Subject: [PATCH] update --- generator/ctrl/training_utils/training.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/generator/ctrl/training_utils/training.py b/generator/ctrl/training_utils/training.py index 4b9c787..5402277 100644 --- a/generator/ctrl/training_utils/training.py +++ b/generator/ctrl/training_utils/training.py @@ -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)