Hypernetworks - fix KeyError in statistics caching

Statistics logging has changed to {filename : list[losses]}, so it has to use loss_info[key].pop()
3 years ago · b297cc3324
parent f2cc3f32d5
commit b297cc3324
1 changed files with 2 additions and 2 deletions
--- a/modules/hypernetworks/hypernetwork.py
+++ b/modules/hypernetworks/hypernetwork.py
@ -274,8 +274,8 @@ def log_statistics(loss_info:dict, key, value):
        loss_info[key] = [value]
    else:
        loss_info[key].append(value)
-        if len(loss_info) > 1024:
+        if len(loss_info[key]) > 1024:
-            loss_info.pop(0)
+            loss_info[key].pop(0)
 def statistics(data):