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