fix visual write (#456)

7f628db5 · LielinJiang · GitHub · 6cbd3e50 · 7f628db5
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

ppgan/engine/trainer.py ppgan/engine/trainer.py +1 -1

未找到文件。
--- a/ppgan/engine/trainer.py
+++ b/ppgan/engine/trainer.py
@@ -204,7 +204,7 @@ class Trainer:
                reader_cost_averager.reset()
                batch_cost_averager.reset()

-            if self.current_iter % self.visual_interval == 0:
+            if self.current_iter % self.visual_interval == 0 and self.local_rank == 0:
                self.visual('visual_train')

            self.learning_rate_scheduler_step()