memory leak fix

This commit is contained in:
2023-06-12 13:59:22 +00:00
parent 874dec0278
commit 33b4ccfcb4
+2 -2
View File
@@ -11,9 +11,9 @@ class Accumulator():
def update(self, dict): def update(self, dict):
for key in dict.keys(): for key in dict.keys():
if key in self.dict.keys(): if key in self.dict.keys():
self.dict[key].append(dict[key]) self.dict[key].append(dict[key].detach().cpu().numpy())
else: else:
self.dict[key] = [dict[key]] self.dict[key] = [dict[key].detach().cpu().numpy()]
def get(self): def get(self):
result = dict() result = dict()