Stick requirement version of astroid to be 2.4.2. Also fixed pylint issue with astroid==2.5 (#4365)
This commit is contained in:
1 parent
d383bc0d83
commit
f8f60b2d83
2 files changed
+3
-2
No files matched your search
@@ -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,4 +1,5 @@
|
||||
coverage==4.4.2
|
||||
astroid==2.4.2
|
||||
pylint==2.5.0
|
||||
pyenchant==1.6.11
|
||||
test-generator==0.1.1
|
||||
|
||||
Reference in new issue
Block a user