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

[POT] Check type for layers' renaming (#9276)

Browse Source 
* change in installation

* develop mode

* change in install guide

* style change

* change declare

* add type checking

* revert changes

* rename directly in nx_model

* Update README_dev.md
This commit is contained in:
Milana Shhanukova
2021-12-28 19:08:36 +04:00
committed by GitHub
parent b2aff0cd56
commit 3cef513495
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tools/pot/openvino/tools/pot/graph/nx_model.py
View File

@@ -186,7 +186,7 @@ class NXModel:
for model_dict in self._models:
model_name, model = model_dict['name'], model_dict['model']
for node in ge.get_all_operation_nodes(model, recursively=False):
rename_node(node, f'{model_name}_{node.name}')
node.name = f'{model_name}_{node.name}'
def _remove_models_prefix(self):
"""Removes model name prefix from node names"""