experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Add inferred output shapes to the GraphDef written by the Supervisor.

Browse Source 
Change: 121885139
This commit is contained in:
A. Unique TensorFlower 2016-05-09 14:37:04 -08:00 committed by TensorFlower Gardener
parent c5ced290a6
commit 9efc4097cf
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/python/training/supervisor.py
View File

@ -575,7 +575,7 @@ class Supervisor(object):
"""Writes graph_def to `logdir` and adds it to summary if applicable."""
assert self._is_chief
if self._logdir:
training_util.write_graph(self._graph.as_graph_def(),
training_util.write_graph(self._graph.as_graph_def(add_shapes=True),
self._logdir, "graph.pbtxt")
if self._summary_writer:
self._summary_writer.add_graph(self._graph)