Commit 88796373 authored by Yuxin Wu's avatar Yuxin Wu

bugfix in DummyConstantInput

parent d263818b
...@@ -398,8 +398,8 @@ class DummyConstantInput(TensorInput): ...@@ -398,8 +398,8 @@ class DummyConstantInput(TensorInput):
tlist = [] tlist = []
ctx = get_current_tower_context() ctx = get_current_tower_context()
assert ctx is not None assert ctx is not None
assert len(self.shapes) == len(self._desc) assert len(self.shapes) == len(self._spec)
for idx, p in enumerate(self._desc): for idx, p in enumerate(self._spec):
tlist.append(tf.constant( tlist.append(tf.constant(
0, dtype=p.dtype, 0, dtype=p.dtype,
name='dummy-{}-{}'.format(p.name, ctx.index), name='dummy-{}-{}'.format(p.name, ctx.index),
......
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