From 5559d1b711dfef64e0b5e3064f2489544f40ba3c Mon Sep 17 00:00:00 2001 From: DKnight54 <126916963+DKnight54@users.noreply.github.com> Date: Fri, 31 Jan 2025 01:56:30 +0800 Subject: [PATCH] Update train_network.py --- train_network.py | 1 + 1 file changed, 1 insertion(+) diff --git a/train_network.py b/train_network.py index 2554b230..5e579451 100644 --- 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