From b2b51e1c76093fed9d444dabc518e50b8ee8d2c5 Mon Sep 17 00:00:00 2001 From: John Pope Date: Wed, 4 Sep 2024 07:00:22 +1000 Subject: [PATCH] ok --- train.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/train.py b/train.py index 44710cf..800a86d 100644 --- a/train.py +++ b/train.py @@ -203,11 +203,11 @@ def train_step(self, x_current, x_reference, global_step): if self.ema: self.ema.update() - # Logging + # Logging - locally for sanity check if global_step % self.config.logging.sample_every == 0: - save_image(x_reconstructed, f'x_reconstructed_{global_step}.png', normalize=True) - save_image(x_current, f'x_current_{global_step}.png', normalize=True) - save_image(x_reference, f'x_reference_{global_step}.png', normalize=True) + save_image(x_reconstructed, f'x_reconstructed.png', normalize=True) + save_image(x_current, f'x_current.png', normalize=True) + save_image(x_reference, f'x_reference.png', normalize=True) return d_loss_avg, g_loss.item(), l_p.item(), l_v.item(), g_loss_gan.item(), x_reconstructed