diff --git a/egs/librispeech/ASR/conformer_ctc/conformer.py b/egs/librispeech/ASR/conformer_ctc/conformer.py index 22d9d8193..871712a46 100644 --- a/egs/librispeech/ASR/conformer_ctc/conformer.py +++ b/egs/librispeech/ASR/conformer_ctc/conformer.py @@ -878,7 +878,7 @@ class ConvolutionModule(nn.Module): groups=channels, bias=bias, ) - if self.use_bathnorm: + if self.use_batchnorm: self.norm = nn.BatchNorm1d(channels) self.pointwise_conv2 = nn.Conv1d( channels, diff --git a/egs/librispeech/ASR/conformer_ctc/transformer.py b/egs/librispeech/ASR/conformer_ctc/transformer.py index 94c0cabf7..086d1f2b8 100644 --- a/egs/librispeech/ASR/conformer_ctc/transformer.py +++ b/egs/librispeech/ASR/conformer_ctc/transformer.py @@ -15,7 +15,7 @@ # limitations under the License. import math -from typing import Dict, List, Optional, Tuple +from typing import Dict, List, Optional, Tuple, Union import torch import torch.nn as nn