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

loss表示追加

Browse Source
This commit is contained in:
u-haru
2023-06-03 21:20:26 +09:00
parent 7c38c33ed6
commit 5907bbd9de
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
train_network.py
View File

@@ -724,7 +724,7 @@ def train(args):
progress_bar.set_postfix(**logs) progress_bar.set_postfix(**logs)
if args.scale_weight_norms: if args.scale_weight_norms:
progress_bar.set_postfix(**max_mean_logs) progress_bar.set_postfix(**{**max_mean_logs,**logs})
if args.logging_dir is not None: if args.logging_dir is not None:
logs = generate_step_logs(args, current_loss, avr_loss, lr_scheduler, keys_scaled, mean_norm, maximum_norm) logs = generate_step_logs(args, current_loss, avr_loss, lr_scheduler, keys_scaled, mean_norm, maximum_norm)