mrq/stable-diffusion-webui
1
1
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update hypernetwork.py

Browse Source
This commit is contained in:
DepFA 2022-10-23 05:43:34 +01:00 committed by AUTOMATIC1111
parent 48dbf99e84
commit 1fbfc052eb
1 changed files with 7 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
modules/hypernetworks/hypernetwork.py
View File

@ -288,10 +288,13 @@ def statistics(data):
def report_statistics(loss_info:dict):
keys = sorted(loss_info.keys(), key=lambda x: sum(loss_info[x]) / len(loss_info[x]))
for key in keys:
info, recent = statistics(loss_info[key])
try:
print("Loss statistics for file " + key)
info, recent = statistics(loss_info[key])
print(info)
print(recent)
except Exception as e:
print(e)