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

[POT] Keep original out names in statistics collection (#14042)

Browse Source
This commit is contained in:
Liubov Talamanova
2022-11-17 13:36:14 +00:00
committed by GitHub
parent 5cf6ceaf83
commit 8d360ea3e6
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tools/pot/openvino/tools/pot/graph/graph_utils.py
View File

@@ -67,7 +67,7 @@ def load_graph(model_config, target_device='ANY'):
return graph_from_ir
def save_graph(graph: Graph, save_path, model_name=None, rename_results=False):
def save_graph(graph: Graph, save_path, model_name=None):
""" Save model as IR in specified path
:param graph: NetworkX model to save
:param save_path: path to save the model
@@ -86,7 +86,7 @@ def save_graph(graph: Graph, save_path, model_name=None, rename_results=False):
graph_copy = deepcopy(graph)
add_removed_converts(graph_copy)
save_restored_graph(graph=graph_copy, path=save_path, meta_data=graph.meta_data,
name=model_name, rename_results=rename_results)
name=model_name)
def model_preprocessing(model):

2
tools/pot/tests/utils/check_graph.py
View File

@@ -16,7 +16,7 @@ def check_graph(tmp_path, graph, model_name, model_framework, check_weights=Fals
model_name = '_'.join([model_name, model_framework])
ir_name_xml = model_name + '.xml'
path_to_ir_xml = tmp_path.joinpath(ir_name_xml)
save_graph(graph, tmp_path.as_posix(), model_name, rename_results=False)
save_graph(graph, tmp_path.as_posix(), model_name)
path_to_ref_ir_xml = REFERENCE_MODELS_PATH.joinpath(ir_name_xml)