From 55fd6e5a35edfc3fa4f3949800ca4622eccc3376 Mon Sep 17 00:00:00 2001 From: Pavel Esir Date: Mon, 10 Aug 2020 09:35:20 +0300 Subject: [PATCH] remove TestMode restriction for batchnorm in Kaldi --- .../mo/front/kaldi/extractors/batchnorm_component_ext.py | 3 --- 1 file changed, 3 deletions(-) diff --git a/model-optimizer/mo/front/kaldi/extractors/batchnorm_component_ext.py b/model-optimizer/mo/front/kaldi/extractors/batchnorm_component_ext.py index 9f94ad638d93e2..09e447a0377bed 100644 --- a/model-optimizer/mo/front/kaldi/extractors/batchnorm_component_ext.py +++ b/model-optimizer/mo/front/kaldi/extractors/batchnorm_component_ext.py @@ -51,9 +51,6 @@ def extract(cls, node): collect_until_token(pb, b'') 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'') mean = read_binary_vector(pb)