From 0d60746ce5c1575804c74f1cbb49c46c3598be48 Mon Sep 17 00:00:00 2001 From: Yves-Noel Weweler Date: Wed, 28 Nov 2018 18:57:17 +0100 Subject: [PATCH] Fix invalid evaluation summary WIP (#3) * The wrong summary was used in the eval metrics. --- tacotron/model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tacotron/model.py b/tacotron/model.py index 86d6be0..f28ae8e 100644 --- a/tacotron/model.py +++ b/tacotron/model.py @@ -495,7 +495,7 @@ def model_fn(self, features, labels, mode, params): ) elif mode == tf.estimator.ModeKeys.EVAL: eval_metrics_ops = { - 'mean_loss': tf.metrics.mean(self.loss_op_decoder, + 'mean_loss': tf.metrics.mean(self.loss_op, name='mean_loss'), 'mean_decoder_loss': tf.metrics.mean(self.loss_op_decoder, name='mean_decoder_loss'),