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

Merge pull request #562 from u-haru/hotfix/max_mean_logs_with_loss

Browse Source 
loss表示追加
This commit is contained in:
Kohya S
2023-06-05 21:42:25 +09:00
committed by GitHub
parent 7c38c33ed6 5907bbd9de
commit 0fe1afd4ef
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)
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:
logs = generate_step_logs(args, current_loss, avr_loss, lr_scheduler, keys_scaled, mean_norm, maximum_norm)