memory leak fix
This commit is contained in:
+2
-2
@@ -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()
|
||||||
|
|||||||
Reference in New Issue
Block a user