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

Minor fixes

Browse Source
This commit is contained in:
pkufool 2021-12-10 15:30:00 +08:00
parent db924dcef5
commit 6dec4b2d8a
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
egs/librispeech/ASR/conformer_ctc/conformer.py
View File

@ -878,7 +878,7 @@ class ConvolutionModule(nn.Module):
groups=channels, groups=channels,
bias=bias, bias=bias,
) )
if self.use_bathnorm: if self.use_batchnorm:
self.norm = nn.BatchNorm1d(channels) self.norm = nn.BatchNorm1d(channels)
self.pointwise_conv2 = nn.Conv1d( self.pointwise_conv2 = nn.Conv1d(
channels, channels,

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

@ -15,7 +15,7 @@
# limitations under the License. # limitations under the License.
import math import math
from typing import Dict, List, Optional, Tuple from typing import Dict, List, Optional, Tuple, Union
import torch import torch
import torch.nn as nn import torch.nn as nn