fix memory leak

Atomu2014 · Oct 9, 2018 · 99d214f · 99d214f
1 parent e832d66
commit 99d214f
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/pnn/tf_main.py b/pnn/tf_main.py
@@ -458,6 +458,7 @@ def train_batch(self, batch_xs, batch_ys):
             _loss_ = np.mean([ret[i] for i in range(1, len(ret), 3)])
             _log_loss_ = np.mean([ret[i] for i in range(2, len(ret), 3)])
             _l2_loss_ = np.mean([ret[i] for i in range(3, len(ret), 3)])
+        del train_feed
         return _loss_, _log_loss_, _l2_loss_
 
     def train(self):
@@ -619,6 +620,7 @@ def evaluate_batch(self, batch_xs, batch_ys):
                     feed_dict[model.training] = False
             _preds_ = self.sess.run(fetches=fetches, feed_dict=feed_dict)
             batch_preds = [x.flatten() for x in _preds_]
+        del feed_dict
         return batch_preds
 
     def evaluate(self, gen, writer=None, eps=1e-6, submission=0):