From ec762fd3afaa6df0715fa1cbe9e6f088b9276ea0 Mon Sep 17 00:00:00 2001
From: Volpeon <git@volpeon.ink>
Date: Fri, 28 Apr 2023 16:22:06 +0200
Subject: Fixed loss/acc logging

---
 training/util.py | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

(limited to 'training/util.py')

diff --git a/training/util.py b/training/util.py
index 61f1533..0b6bea9 100644
--- a/training/util.py
+++ b/training/util.py
@@ -1,5 +1,6 @@
 from pathlib import Path
 import json
+import math
 from typing import Iterable, Any
 from contextlib import contextmanager
 
@@ -23,7 +24,9 @@ class AverageMeter:
 
     def reset(self):
         self.step = 0
-        self.avg = 0
+        self.min = math.inf
+        self.max = 0.0
+        self.avg = 0.0
 
     def get_decay(self):
         if self.step <= 0:
@@ -35,6 +38,8 @@ class AverageMeter:
         for _ in range(n):
             self.step += n
             self.avg += (val - self.avg) * self.get_decay()
+            self.min = min(self.min, self.avg)
+            self.max = max(self.max, self.avg)
 
 
 class EMAModel(EMAModel_):
-- 
cgit v1.2.3-54-g00ecf