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: simplify validation step condition in NetworkTrainer

Browse Source
This commit is contained in:
Kohya S
2025-02-11 21:53:57 +09:00
parent cd80752175
commit 76b761943b
1 changed files with 3 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
train_network.py
View File

@@ -1414,12 +1414,9 @@ class NetworkTrainer:
) )
accelerator.log(logs, step=global_step) accelerator.log(logs, step=global_step)
# VALIDATION PER STEP # VALIDATION PER STEP: global_step is already incremented
should_validate_step = ( # for example, if validate_every_n_steps=100, validate at step 100, 200, 300, ...
args.validate_every_n_steps is not None should_validate_step = args.validate_every_n_steps is not None and global_step % args.validate_every_n_steps == 0
and global_step != 0 # Skip first step
and global_step % args.validate_every_n_steps == 0
)
if accelerator.sync_gradients and validation_steps > 0 and should_validate_step: if accelerator.sync_gradients and validation_steps > 0 and should_validate_step:
optimizer_eval_fn() optimizer_eval_fn()
accelerator.unwrap_model(network).eval() accelerator.unwrap_model(network).eval()