From d425daad5b383007b3e9d40bf591ebe2f75ea008 Mon Sep 17 00:00:00 2001
From: erogol <erogol@hotmail.com>
Date: Mon, 1 Jun 2020 03:02:51 +0200
Subject: [PATCH] tb eval print

---
 vocoder/train.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/vocoder/train.py b/vocoder/train.py
index a30e3bff..27e14e8a 100644
--- a/vocoder/train.py
+++ b/vocoder/train.py
@@ -383,6 +383,7 @@ def evaluate(model_G, criterion_G, model_D, ap, global_step, epoch):
     # synthesize a full voice
     data_loader.return_segments = False
 
+    tb_logger.tb_eval_stats(global_step, keep_avg.avg_values)
     return keep_avg.avg_values