From 33b4ccfcb4fb72b4a92f9e9ae42cdb64a6e4dc21 Mon Sep 17 00:00:00 2001 From: Hunor Laczko Date: Mon, 12 Jun 2023 13:59:22 +0000 Subject: [PATCH] memory leak fix --- utils/utils.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/utils/utils.py b/utils/utils.py index 337bc40..ef842cd 100644 --- a/utils/utils.py +++ b/utils/utils.py @@ -11,9 +11,9 @@ class Accumulator(): def update(self, dict): for key in dict.keys(): if key in self.dict.keys(): - self.dict[key].append(dict[key]) + self.dict[key].append(dict[key].detach().cpu().numpy()) else: - self.dict[key] = [dict[key]] + self.dict[key] = [dict[key].detach().cpu().numpy()] def get(self): result = dict()