Merge pull request #1937 from jazcollins/master

removing memory intensive session.graph from FileWriter call

Merge pull request #1937 from jazcollins/master
removing memory intensive session.graph from FileWriter call
ce840bf0 · David Sussillo · GitHub · b879a927 · 8adbb64c · ce840bf0
Commit ce840bf0 authored Jul 12, 2017 by David Sussillo Committed by GitHub Jul 12, 2017
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

lfads/lfads.py lfads/lfads.py +1 -1

No files found.
--- a/lfads/lfads.py
+++ b/lfads/lfads.py
@@ -952,7 +952,7 @@ class LFADS(object):
    session = tf.get_default_session()
    self.logfile = os.path.join(hps.lfads_save_dir, "lfads_log")
-    self.writer = tf.summary.FileWriter(self.logfile, session.graph)
+    self.writer = tf.summary.FileWriter(self.logfile)
  def build_feed_dict(self, train_name, data_bxtxd, ext_input_bxtxi=None,
                      keep_prob=None):