diff --git a/egs/librispeech/ASR/conformer_ctc/transformer.py b/egs/librispeech/ASR/conformer_ctc/transformer.py index 52bf4a313..00ca027a7 100644 --- a/egs/librispeech/ASR/conformer_ctc/transformer.py +++ b/egs/librispeech/ASR/conformer_ctc/transformer.py @@ -76,6 +76,7 @@ class Transformer(nn.Module): """ super().__init__() self.use_feat_batchnorm = use_feat_batchnorm + assert isinstance(use_feat_batchnorm, (float, bool)) if isinstance(use_feat_batchnorm, bool) and use_feat_batchnorm: self.feat_batchnorm = nn.BatchNorm1d(num_features)