Commit c223f223 authored by Yuxin Wu's avatar Yuxin Wu

fix build

parent 4f498c2a
......@@ -106,7 +106,8 @@ class TowerContext(object):
for c in self._ctxs:
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()
assert ns == self._name, \
"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