From 83a171d3dc304905c493effb63bfd0f85f909190 Mon Sep 17 00:00:00 2001
From: DKnight54 <126916963+DKnight54@users.noreply.github.com>
Date: Sun, 2 Feb 2025 04:56:22 +0800
Subject: [PATCH] Update train_network.py

---
 train_network.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/train_network.py b/train_network.py
index 17a3b36e..2accc40a 100644
--- a/train_network.py
+++ b/train_network.py
@@ -1026,8 +1026,8 @@ class NetworkTrainer:
                 # Collecting latents and caption lists from all processes
                 all_latents = gather_object([latents])
                 all_captions =  gather_object(batch["captions"])
-                logger.info(f"latents: {latents}")
-                logger.info(f"all_latents: {all_latents}")
+                #logger.info(f"latents: {latents}")
+                #logger.info(f"all_latents: {all_latents}")
                 example_tuple = (all_latents, all_captions)
                 if accelerator.sync_gradients:
                     progress_bar.update(1)