From 7e9fc986098bd821ebd6e47009941fbcad80c0be Mon Sep 17 00:00:00 2001
From: Bobholamovic <bob1998425@hotmail.com>
Date: Thu, 26 Mar 2020 21:52:06 +0800
Subject: [PATCH] Fix confusion matrix isn't reset
---
src/utils/metrics.py | 5 +++++
1 file changed, 5 insertions(+)
diff --git a/src/utils/metrics.py b/src/utils/metrics.py
index 3e2b24f..3a13775 100644
--- a/src/utils/metrics.py
+++ b/src/utils/metrics.py
@@ -23,6 +23,11 @@ class AverageMeter:
self.sum = 0
self.count = 0
+ for attr in filter(lambda a: not a.startswith('__'), dir(self)):
+ obj = getattr(self, attr)
+ if isinstance(obj, AverageMeter):
+ AverageMeter.reset(obj)
+
def update(self, *args, n=1):
self.val = self.compute(*args)
self.sum += self.val * n
--
GitLab