From 7fcf409fbcc512c6ab2af19ce67e92bff1eb06ec Mon Sep 17 00:00:00 2001 From: zuoxiang95 Date: Thu, 16 Nov 2017 17:19:02 +0800 Subject: [PATCH] fix --- models.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/models.py b/models.py index 7026125..5e63268 100644 --- a/models.py +++ b/models.py @@ -188,7 +188,7 @@ def build(self): def get_loss(self): if (self.config.opt == 'Adam'): optimizer = tf.train.AdamOptimizer(self.config.lr) - elif (self.config.opt == 'Adagrad'): + else: optimizer = tf.train.AdagradOptimizer(self.config.lr) with tf.variable_scope("Model"): self.loss = tf.nn.sparse_softmax_cross_entropy_with_logits(logits=self.final_output,