Restore change of i64 parameters to i32 (#1866)

2020-08-19 20:18:04 +03:00
parent a8e611f90b
commit 8a6b3b6d89
1 changed files with 4 additions and 0 deletions
--- a/model-optimizer/extensions/front/ChangePlaceholderTypes.py
+++ b/model-optimizer/extensions/front/ChangePlaceholderTypes.py
@@ -48,6 +48,10 @@ class ChangePlaceholderTypes(FrontReplacementPattern):
                            op.out_port(0).connect(dst_port)

                        graph.remove_node(convert_node.id)
+            if op.soft_get('data_type') == np.int64:
+                op.data_type = np.int32
+                log.error('Convert data type of Parameter "{}" to int32'.format(op.soft_get('name', op.id)),
+                          extra={'is_warning': True})

            if op.soft_get('data_type') == np.uint8:
                op.data_type = np.float32