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