Created
March 29, 2022 11:27
-
-
Save daniil-lyakhov/8bce5bebb043f8131976e8287c7e5c53 to your computer and use it in GitHub Desktop.
Openvino Pruning Transformation on IR
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
import sys | |
from openvino.frontend import FrontEndManager | |
from openvino.offline_transformations import apply_pruning_transformation | |
from openvino.runtime import serialize | |
print(f'input model path {sys.argv[1]} \n output model path {sys.argv[2]}.xml \n output model weights {sys.argv[2]}.bin') | |
input_model = sys.argv[1] | |
fem = FrontEndManager() | |
def read_model(path_to_xml): | |
fe = fem.load_by_framework(framework="ir") | |
function = fe.convert(fe.load(path_to_xml)) | |
return function | |
func = read_model(input_model) | |
apply_pruning_transformation(func) | |
serialize(func, sys.argv[2] + '.xml', sys.argv[2] + '.bin') | |
print('Done!') |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment