Update train_network.py

2026-04-17 09:18:00 +00:00 · 2025-02-02 04:39:15 +08:00
parent 6eabaf3775
commit 6fc38477c8
1 changed files with 1 additions and 0 deletions
--- a/train_network.py
+++ b/train_network.py
@@ -1026,6 +1026,7 @@ class NetworkTrainer:
                # Collecting latents and caption lists from all processes
                all_latents = gather_object(latents)
                all_captions =  gather_object(batch["captions"])
+                logger.log(f"all_latents: {all_latents}")
                example_tuple = (all_latents, all_captions)
                if accelerator.sync_gradients:
                    progress_bar.update(1)