iicd/xautodl
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Avoid the memory leak

Browse Source
This commit is contained in:
D-X-Y 2020-03-21 16:04:12 +11:00
parent e76451c791
commit d8784b3070
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
lib/nas_201_api/api.py
View File

@ -735,6 +735,7 @@ class ArchResults(object):
# This can help reduce the memory footprint. # This can help reduce the memory footprint.
def clear_params(self): def clear_params(self):
for key, result in self.all_results.items(): for key, result in self.all_results.items():
del result.net_state_dict
result.net_state_dict = None result.net_state_dict = None
self.clear_net_done = True self.clear_net_done = True