Update hypernetwork.py

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

View File

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