Fixed the VAE validation images being sent too many times to wandb du…

…ring validation causing network issues because of it hitting the api rate limit.
ZeroCool940711 · Sep 5, 2023 · fce8fd1 · fce8fd1
1 parent aaddf6e
commit fce8fd1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/muse_maskgit_pytorch/trainers/vqvae_trainers.py b/muse_maskgit_pytorch/trainers/vqvae_trainers.py
@@ -236,7 +236,7 @@ def log_validation_images(self, logs, steps):
                 log_imgs.append(Image.open(ema_save_path))
                 prompts.append("ema")
 
-            super().log_validation_images(log_imgs, steps, prompts=prompts)
+        super().log_validation_images(log_imgs, steps, prompts=prompts)
         self.model.train()
 
     def train(self):