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

fix calculation of memoryoffset shape (#4533)

Browse Source 
* fix calculation of memoryoffset shape in 2 cases: after normalization layer (shape can be read from InputDim in loader) and after affincecomponent layer (shape saved in out-size based on weights shape)

* improve comments
This commit is contained in:
Svetlana Dolinina 2021-03-02 14:35:22 +03:00 committed by GitHub
parent 473c944e6e
commit 42c1c65454
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 20 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
model-optimizer/extensions/front/kaldi/split_recurrent_memoryoffset.py
View File

@ -1,5 +1,5 @@
""" """
Copyright (C) 2018-2020 Intel Corporation Copyright (C) 2018-2021 Intel Corporation
Licensed under the Apache License, Version 2.0 (the "License"); Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License. you may not use this file except in compliance with the License.
@ -57,6 +57,13 @@ class SplitRecurrentMemoryOffset(FrontReplacementSubgraph):
# MemoryOffset node is not in a recurrent block -- no splitting is needed # MemoryOffset node is not in a recurrent block -- no splitting is needed
return return
# check that node has information for future partial infer
# element_size is set in loader based on dimensions of previous layer from original Kaldi model
if not offset_node.has_valid('element_size'): if not offset_node.has_valid('element_size'):
# check if previous layer contains information about its shape in out-size
# out-size is set in extractor of some nodes like affinecomponent based on weight's size
if offset_node.in_port(0).get_source().node.has_valid('out-size'):
offset_node['element_size'] = offset_node.in_port(0).get_source().node['out-size']
else:
raise Error("In a recurrent block 'element_size' for node {} is not set".format(offset_node.id)) raise Error("In a recurrent block 'element_size' for node {} is not set".format(offset_node.id))
SplitRecurrentMemoryOffset.split_offset(offset_node) SplitRecurrentMemoryOffset.split_offset(offset_node)

5
model-optimizer/mo/front/kaldi/loader/loader.py
View File

@ -234,11 +234,14 @@ def load_components(file_descr, graph, component_layer_map=None):
# it is separated in 2 parts to remove cycle from graph # it is separated in 2 parts to remove cycle from graph
file_descr.seek(start_index) file_descr.seek(start_index)
dim = 0 dim = 0
dim_words = {b'<Dim>', b'<InputDim>'}
for dim_word in dim_words:
try: try:
collect_until_token(file_descr, b'<Dim>', size_search_zone=end_index - start_index) collect_until_token(file_descr, dim_word, size_search_zone=end_index - start_index)
cur_index = file_descr.tell() cur_index = file_descr.tell()
if start_index < cur_index < end_index: if start_index < cur_index < end_index:
dim = read_binary_integer32_token(file_descr) dim = read_binary_integer32_token(file_descr)
break
else: else:
file_descr.seek(start_index) file_descr.seek(start_index)
except Error: except Error: