bug fix

2020-11-10 12:53:39 +01:00 · 2020-11-10 12:53:39 +01:00 · f8fd300b3e
parent 016d3503da
commit f8fd300b3e
1 changed files with 1 additions and 1 deletions
--- a/TTS/bin/train_vocoder_wavegrad.py
+++ b/TTS/bin/train_vocoder_wavegrad.py
@ -221,7 +221,7 @@ def train(model, criterion, optimizer,
    if args.rank == 0:
        tb_logger.tb_train_epoch_stats(global_step, epoch_stats)
    # TODO: plot model stats
-    if c.tb_model_param_stats:
+    if c.tb_model_param_stats and args.rank == 0:
        tb_logger.tb_model_weights(model, global_step)
    return keep_avg.avg_values, global_step