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

slyalin · Jun 4, 2020 · c7d130e · c7d130e
1 parent c10ff28
commit c7d130e
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/model-optimizer/extensions/back/ProposalMutation.py b/model-optimizer/extensions/back/ProposalMutation.py
@@ -83,4 +83,6 @@ def replace_pattern(graph: Graph, match: dict):
             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'