Skip to content

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

bugfix

parent 6262f719
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
tensorpack/tfutils/tower.py
View file @ 4dadc6f0
...@@ -372,9 +372,9 @@ class TowerTensorHandle(object): ...@@ -372,9 +372,9 @@ class TowerTensorHandle(object):
self._extra_tensor_names = {} self._extra_tensor_names = {}
if input_signature is not None: if input_signature is not None:
assert len(input_signature) == len(input) assert len(input_signature) == len(inputs)
self._extra_tensor_names = { self._extra_tensor_names = {
get_op_tensor_name(x.name)[1]: y for x, y in zip(input_signature, input)} get_op_tensor_name(x.name)[1]: y for x, y in zip(input_signature, inputs)}
self._inputs = inputs self._inputs = inputs
self._outputs = outputs self._outputs = outputs
......
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