mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-08-27 02:34:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add assertion for use_feat_batchnorm

Browse Source
This commit is contained in:
pkufool 2021-12-14 08:11:44 +08:00
parent 5bfcf65cca
commit 67ed6225a2
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
egs/librispeech/ASR/conformer_ctc/transformer.py
View File

@ -76,6 +76,7 @@ class Transformer(nn.Module):
""" """
super().__init__() super().__init__()
self.use_feat_batchnorm = use_feat_batchnorm self.use_feat_batchnorm = use_feat_batchnorm
assert isinstance(use_feat_batchnorm, (float, bool))
if isinstance(use_feat_batchnorm, bool) and use_feat_batchnorm: if isinstance(use_feat_batchnorm, bool) and use_feat_batchnorm:
self.feat_batchnorm = nn.BatchNorm1d(num_features) self.feat_batchnorm = nn.BatchNorm1d(num_features)