From 56be1a54388ec660b7279fcf71493501471e1ec6 Mon Sep 17 00:00:00 2001 From: Daniil Lyakhov Date: Mon, 14 Feb 2022 15:06:09 +0300 Subject: [PATCH] Change User Transformations applying order in MO (#10241) * Fix user transformation order in mo * Move user transformation behind FP16 compression * Move user transformation call before fp16 compression --- tools/mo/openvino/tools/mo/back/offline_transformations.py | 2 +- tools/mo/openvino/tools/mo/moc_frontend/serialize.py | 3 ++- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/tools/mo/openvino/tools/mo/back/offline_transformations.py b/tools/mo/openvino/tools/mo/back/offline_transformations.py index bc26d324bc9..b2d18e55950 100644 --- a/tools/mo/openvino/tools/mo/back/offline_transformations.py +++ b/tools/mo/openvino/tools/mo/back/offline_transformations.py @@ -94,11 +94,11 @@ def apply_offline_transformations(input_model: str, argv: argparse.Namespace): if 'scale' in argv: argv.scale = scale - apply_user_transformations(func, parse_transform(argv.transform)) apply_moc_transformations(func) params_with_custom_types = create_params_with_custom_types(argv.packed_user_shapes) apply_moc_legacy_transformations(func, params_with_custom_types) + apply_user_transformations(func, parse_transform(argv.transform)) if "compress_fp16" in argv and argv.compress_fp16: compress_model(func) diff --git a/tools/mo/openvino/tools/mo/moc_frontend/serialize.py b/tools/mo/openvino/tools/mo/moc_frontend/serialize.py index 1268293c782..488c0b9120f 100644 --- a/tools/mo/openvino/tools/mo/moc_frontend/serialize.py +++ b/tools/mo/openvino/tools/mo/moc_frontend/serialize.py @@ -22,7 +22,6 @@ def moc_emit_ir(ngraph_function: Model, argv: argparse.Namespace): from openvino.tools.mo.back.offline_transformations import apply_user_transformations, apply_moc_transformations, \ apply_moc_legacy_transformations - apply_user_transformations(ngraph_function, parse_transform(argv.transform)) apply_moc_transformations(ngraph_function) from openvino.offline_transformations import compress_quantize_weights_transformation compress_quantize_weights_transformation(ngraph_function) @@ -33,6 +32,8 @@ def moc_emit_ir(ngraph_function: Model, argv: argparse.Namespace): else list(argv.placeholder_data_types.keys()) apply_moc_legacy_transformations(ngraph_function, params_with_custom_types) + apply_user_transformations(ngraph_function, parse_transform(argv.transform)) + if argv.compress_fp16: from openvino.tools.mo.back.offline_transformations import compress_model compress_model(ngraph_function)