Skip to content

S
seminar-breakout
You need to sign in or sign up before continuing.
Commit e9da8f89 authored by Yuxin Wu's avatar Yuxin Wu
Browse files
Options

fix bug in get_op_tensor_name. Make sample in GAN examples work.

parent 562f0906
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tensorpack/tfutils/common.py
View file @ e9da8f89
...@@ -81,7 +81,7 @@ def get_op_tensor_name(name): ...@@ -81,7 +81,7 @@ def get_op_tensor_name(name):
Returns: Returns:
tuple: (op_name, tensor_name) tuple: (op_name, tensor_name)
""" """
if name[-2] == ':': if len(name) >= 3 and name[-2] == ':':
return name[:-2], name return name[:-2], name
else: else:
return name, name + ':0' return name, name + ':0'
......
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