memory leak fix

2023-06-12 13:59:22 +00:00
parent 874dec0278
commit 33b4ccfcb4
1 changed files with 2 additions and 2 deletions
@@ -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()