Skip to content

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

fix build_graph args (#459)

parent 5b5d203f
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tensorpack/graph_builder/model_desc.py
View file @ 118aae66
...@@ -98,7 +98,7 @@ class ModelDescBase(object): ...@@ -98,7 +98,7 @@ class ModelDescBase(object):
arg = args[0] arg = args[0]
if isinstance(arg, InputSource): if isinstance(arg, InputSource):
inputs = arg.get_input_tensors() # remove in the future? inputs = arg.get_input_tensors() # remove in the future?
if isinstance(arg, (list, tuple)): elif isinstance(arg, (list, tuple)):
inputs = arg inputs = arg
else: else:
inputs = [arg] inputs = [arg]
......
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