Update train_network.py

2026-04-15 16:39:42 +00:00 · 2025-01-31 01:56:30 +08:00
parent 12dd9f6b59
commit 5559d1b711
1 changed files with 1 additions and 0 deletions
--- a/train_network.py
+++ b/train_network.py
@@ -1023,6 +1023,7 @@ class NetworkTrainer:

                # Checks if the accelerator has performed an optimization step behind the scenes
                example_tuple = (latents, batch["captions"])
+                logger.info(f"Example tuple: {example_tuple}")
                if accelerator.sync_gradients:
                    progress_bar.update(1)
                    global_step += 1