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

Change User Transformations applying order in MO (#10241)

Browse Source 
* Fix user transformation order in mo

* Move user transformation behind FP16 compression

* Move user transformation call before fp16 compression
This commit is contained in:
Daniil Lyakhov 2022-02-14 15:06:09 +03:00 committed by GitHub
parent a9b6eaf5c0
commit 56be1a5438
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tools/mo/openvino/tools/mo/back/offline_transformations.py
View File

@ -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)

3
tools/mo/openvino/tools/mo/moc_frontend/serialize.py
View File

@ -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)