IntenseWebs/openvino
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix Proposal for the case of 2 outputs (#773)

Browse Source
This commit is contained in:
Maxim Vafin 2020-06-04 20:56:46 +03:00 committed by GitHub
parent c10ff28f12
commit c7d130efbe
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
model-optimizer/extensions/back/ProposalMutation.py
View File

@ -83,4 +83,6 @@ class ProposalMutation(BackReplacementPattern):
reshape.out_port(0).connect(node.in_port(2))
if node.has_port('out', 1) and not node.out_port(1).disconnected():
node['version'] = 'extension'
# This is the case when Proposal layer is used from extension, not from opset.
# Setting version attribute is not recommended, this will be fixed after Proposal will be updated in IE.
graph.node[node.id]['version'] = 'extension'