Skip to content

GitLab

Commit 8adbb64c authored by Jasmine's avatar Jasmine
Browse files

removing memory intensive session.graph from FileWriter call

parent b879a927
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
lfads/lfads.py
View file @ 8adbb64c
...@@ -952,7 +952,7 @@ class LFADS(object): ...@@ -952,7 +952,7 @@ class LFADS(object):
session = tf.get_default_session() session = tf.get_default_session()
self.logfile = os.path.join(hps.lfads_save_dir, "lfads_log") 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, def build_feed_dict(self, train_name, data_bxtxd, ext_input_bxtxi=None,
keep_prob=None): keep_prob=None):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment