Merge pull request dennybritz#120 from RoshanTanisha/master

fixed memory leak
Beokane · Dec 4, 2017 · 4e24742 · 4e24742
2 parents c0ae99c + 6de4fee
commit 4e24742
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/train.py b/train.py
@@ -65,6 +65,9 @@
 dev_sample_index = -1 * int(FLAGS.dev_sample_percentage * float(len(y)))
 x_train, x_dev = x_shuffled[:dev_sample_index], x_shuffled[dev_sample_index:]
 y_train, y_dev = y_shuffled[:dev_sample_index], y_shuffled[dev_sample_index:]
+
+del x, y, x_shuffled, y_shuffled
+
 print("Vocabulary Size: {:d}".format(len(vocab_processor.vocabulary_)))
 print("Train/Dev split: {:d}/{:d}".format(len(y_train), len(y_dev)))