Avoid the memory leak
This commit is contained in:
parent
e76451c791
commit
d8784b3070
@ -735,6 +735,7 @@ class ArchResults(object):
|
||||
# This can help reduce the memory footprint.
|
||||
def clear_params(self):
|
||||
for key, result in self.all_results.items():
|
||||
del result.net_state_dict
|
||||
result.net_state_dict = None
|
||||
self.clear_net_done = True
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user