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

Stick requirement version of astroid to be 2.4.2. Also fixed pylint issue with astroid==2.5 (#4365)

Browse Source
This commit is contained in:
Evgeny Lazarev 2021-02-16 21:34:51 +03:00 committed by GitHub
parent d383bc0d83
commit f8f60b2d83
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

4
model-optimizer/extensions/middle/RNNSequenceNormalizeToIE.py
View File

@ -34,7 +34,7 @@ class RNNSequenceNormalize(MiddleReplacementPattern):
This class normalize RNNSequence layers to IE-compatible from of weights, inputs and outputs.
In this pass next will be done:
1. Weights repack (squeeze all useless shapes in all blobls and concatenate W and R together, also add
1. Weights repack (squeeze all useless shapes in all blobs and concatenate W and R together, also add
bin param and all similar staff )
1. UNSqueeze num directions (in states and )
2. Initial states squeeze
@ -104,7 +104,7 @@ class RNNSequenceNormalize(MiddleReplacementPattern):
B_tmp[:, :, :, 3, :] = B[:, :, 1, 2, :][:, :, np.newaxis, :]
B = B_tmp
else:
B = np.add.reduce(B, axis=2, keepdims=True)
B = np.sum(B, axis=2, keepdims=True)
# Concatenate W, R to IE-compatible format
assert len(W.shape) == 5

1
model-optimizer/requirements_dev.txt
View File

@ -1,4 +1,5 @@
coverage==4.4.2
astroid==2.4.2
pylint==2.5.0
pyenchant==1.6.11
test-generator==0.1.1