mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-08-26 18:24:18 +00:00
Add assertion for use_feat_batchnorm
This commit is contained in:
parent
5bfcf65cca
commit
67ed6225a2
@ -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)
|
||||
|
||||
|
Loading…
x
Reference in New Issue
Block a user