From 67ed6225a22370b205646a4ba6c03d319cb06e67 Mon Sep 17 00:00:00 2001 From: pkufool Date: Tue, 14 Dec 2021 08:11:44 +0800 Subject: [PATCH] Add assertion for use_feat_batchnorm --- egs/librispeech/ASR/conformer_ctc/transformer.py | 1 + 1 file changed, 1 insertion(+) 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)