diff --git a/train_network.py b/train_network.py
index ffd1cbbf..4c7d3bf3 100644
--- a/train_network.py
+++ b/train_network.py
@@ -929,7 +929,7 @@ class NetworkTrainer:
                         if torch.any(torch.isnan(latents)):
                             accelerator.print("NaN found in latents, replacing with zeros")
                             latents = torch.nan_to_num(latents, 0, out=latents)
-                        logger.info(f"Latents Shape: {latents.shape}")
+                    logger.info(f"Latents Shape: {latents.shape}")
                     latents = latents * self.vae_scale_factor
 
                     # get multiplier for each sample