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

black formatted

Browse Source
This commit is contained in:
jinzr 2023-10-31 10:23:11 +08:00
parent fde8a2ff65
commit 72d947387d
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
egs/librispeech/ASR/tiny_transducer_ctc/ctc_decode.py
View File

@ -22,10 +22,11 @@
import argparse import argparse
import logging import logging
import math import math
import pprint
from collections import defaultdict from collections import defaultdict
from pathlib import Path from pathlib import Path
from typing import Dict, List, Optional, Tuple from typing import Dict, List, Optional, Tuple
import pprint
import k2 import k2
import sentencepiece as spm import sentencepiece as spm
import torch import torch

1
egs/librispeech/ASR/tiny_transducer_ctc/encoder.py
View File

@ -253,7 +253,6 @@ class CausalSqueezeExcite1d(nn.Module):
return y return y
def forward(self, x: Tensor) -> Tensor: def forward(self, x: Tensor) -> Tensor:
assert len(x.shape) == 3, "Input is not a 3D tensor!" assert len(x.shape) == 3, "Input is not a 3D tensor!"
y = self.exponential_moving_avg(x) y = self.exponential_moving_avg(x)
y = y.permute(0, 2, 1) # make channel last for squeeze op y = y.permute(0, 2, 1) # make channel last for squeeze op