VictorMylle/Kohya-ss-sd-scripts
1
0
Fork 1
mirror of https://github.com/kohya-ss/sd-scripts.git synced 2026-04-08 22:35:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: unpause training progress bar after vaidation

Browse Source
This commit is contained in:
Kohya S
2025-02-11 21:42:46 +09:00
parent 344845b429
commit 177203818a
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
train_network.py
View File

@@ -1489,6 +1489,7 @@ class NetworkTrainer:
args.max_timestep = original_args_max_timestep args.max_timestep = original_args_max_timestep
optimizer_train_fn() optimizer_train_fn()
accelerator.unwrap_model(network).train() accelerator.unwrap_model(network).train()
progress_bar.unpause()
if global_step >= args.max_train_steps: if global_step >= args.max_train_steps:
break break
@@ -1572,6 +1573,7 @@ class NetworkTrainer:
args.max_timestep = original_args_max_timestep args.max_timestep = original_args_max_timestep
optimizer_train_fn() optimizer_train_fn()
accelerator.unwrap_model(network).train() accelerator.unwrap_model(network).train()
progress_bar.unpause()
# END OF EPOCH # END OF EPOCH
if is_tracking: if is_tracking: