Skip to content

S
seminar-breakout
Commit c223f223 authored by Yuxin Wu's avatar Yuxin Wu
Browse files
Options

fix build

parent 4f498c2a
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
tensorpack/tfutils/tower.py
View file @ c223f223
...@@ -106,7 +106,8 @@ class TowerContext(object): ...@@ -106,7 +106,8 @@ class TowerContext(object):
for c in self._ctxs: for c in self._ctxs:
c.__enter__() c.__enter__()
if get_tf_version_number() >= 1.2: # currently only check for predictor towers
if not self.is_training and get_tf_version_number() >= 1.2:
ns = tf.get_default_graph().get_name_scope() ns = tf.get_default_graph().get_name_scope()
assert ns == self._name, \ assert ns == self._name, \
"Name conflict: name_scope inside tower '{}' becomes '{}'!".format(self._name, ns) \ "Name conflict: name_scope inside tower '{}' becomes '{}'!".format(self._name, ns) \
......
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