From 6dec4b2d8ade4a9801ffae4f59173d2eb855755e Mon Sep 17 00:00:00 2001 From: pkufool Date: Fri, 10 Dec 2021 15:30:00 +0800 Subject: [PATCH] Minor fixes --- egs/librispeech/ASR/conformer_ctc/conformer.py | 2 +- egs/librispeech/ASR/conformer_ctc/transformer.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) 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