Add assertion for use_feat_batchnorm

This commit is contained in:
pkufool 2021-12-14 08:11:44 +08:00
parent 5bfcf65cca
commit 67ed6225a2

View File

@ -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)