From c7d130efbec2a3106e24b9073d5c6e4810e1925d Mon Sep 17 00:00:00 2001 From: Maxim Vafin Date: Thu, 4 Jun 2020 20:56:46 +0300 Subject: [PATCH] Fix Proposal for the case of 2 outputs (#773) --- model-optimizer/extensions/back/ProposalMutation.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/model-optimizer/extensions/back/ProposalMutation.py b/model-optimizer/extensions/back/ProposalMutation.py index 1017ef45580..917d0b42e30 100644 --- a/model-optimizer/extensions/back/ProposalMutation.py +++ b/model-optimizer/extensions/back/ProposalMutation.py @@ -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'