mirror of
https://github.com/kohya-ss/sd-scripts.git
synced 2026-04-17 17:24:21 +00:00
Update train_network.py
This commit is contained in:
@@ -1023,7 +1023,6 @@ class NetworkTrainer:
|
|||||||
|
|
||||||
# Checks if the accelerator has performed an optimization step behind the scenes
|
# Checks if the accelerator has performed an optimization step behind the scenes
|
||||||
example_tuple = (latents, batch["captions"])
|
example_tuple = (latents, batch["captions"])
|
||||||
logger.info(f"Example tuple: {example_tuple}")
|
|
||||||
if accelerator.sync_gradients:
|
if accelerator.sync_gradients:
|
||||||
progress_bar.update(1)
|
progress_bar.update(1)
|
||||||
global_step += 1
|
global_step += 1
|
||||||
|
|||||||
Reference in New Issue
Block a user