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

remove TestMode restriction for batchnorm in Kaldi (#1697)

Browse Source
This commit is contained in:
Pavel Esir 2020-08-10 12:21:53 +03:00 committed by GitHub
parent 75d2d88b61
commit 7e82728130
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 0 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
model-optimizer/mo/front/kaldi/extractors/batchnorm_component_ext.py
View File

@ -51,9 +51,6 @@ class BatchNormComponentFrontExtractor(FrontExtractorOp):
collect_until_token(pb, b'<TestMode>')
test_mode = read_binary_bool_token(pb)
if test_mode is not False:
raise Error("Test mode True for BatchNorm is not supported")
collect_until_token(pb, b'<StatsMean>')
mean = read_binary_vector(pb)