Pārlūkot izejas kodu

free gpu memory used

Jérôme BUISINE 4 gadi atpakaļ
vecāks
revīzija
bc65afde40
1 mainītis faili ar 1 papildinājumiem un 0 dzēšanām
  1. 1 0
      find_best_attributes_surrogate_dl.py

+ 1 - 0
find_best_attributes_surrogate_dl.py

@@ -230,6 +230,7 @@ def main():
         test_roc_auc = roc_auc_score(y_test, y_test_predict)
 
         end = datetime.datetime.now()
+        del model
 
         diff = end - start