michael/stable-diffusion-webui
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages 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

11
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])
print("Loss statistics for file " + key)
print(info)
print(recent)
try:
print("Loss statistics for file " + key)
info, recent = statistics(loss_info[key])
print(info)
print(recent)
except Exception as e:
print(e)