tb eval print

2020-06-01 03:02:51 +02:00 · 2020-06-01 03:02:51 +02:00 · d425daad5b
parent 78cb7478d1
commit d425daad5b
1 changed files with 1 additions and 0 deletions
--- 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