removing memory intensive session.graph from FileWriter call

8adbb64c · Jasmine · b879a927 · 8adbb64c
Commit 8adbb64c authored Jul 12, 2017 by Jasmine
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):