Merge pull request #48 from lokitoth/fix/pytorch_graph

Fix graph generation for PyTorch >= 1.2.0
This commit is contained in:
Shital Shah 2019-11-05 09:18:30 -08:00 коммит произвёл GitHub
Родитель 32832b162c b86b1d9c87
Коммит c31cccc844
Не найден ключ, соответствующий данной подписи
Идентификатор ключа GPG: 4AEE18F83AFDEB23
1 изменённых файлов: 1 добавлений и 1 удалений

Просмотреть файл

@ -44,7 +44,7 @@ def pytorch_id(node):
"""Returns a unique ID for a node."""
# After ONNX simplification, the scopeName is not unique anymore
# so append node outputs to guarantee uniqueness
return node.scopeName() + "/outputs/" + "/".join([o.uniqueName() for o in node.outputs()])
return node.scopeName() + "/outputs/" + "/".join([str(o.unique()) for o in node.outputs()])
def get_shape(torch_node):