From d8784b3070cfb938df4ae9a3b99379a9dad75ce1 Mon Sep 17 00:00:00 2001
From: D-X-Y <280835372@qq.com>
Date: Sat, 21 Mar 2020 16:04:12 +1100
Subject: [PATCH] Avoid the memory leak

---
 lib/nas_201_api/api.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/lib/nas_201_api/api.py b/lib/nas_201_api/api.py
index 5a4db25..7d64666 100644
--- a/lib/nas_201_api/api.py
+++ b/lib/nas_201_api/api.py
@@ -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