mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-12-11 06:55:27 +00:00
Remove ZipformerEncoder.norm
This commit is contained in:
parent
0de1184c6d
commit
fbdb12cf77
@ -216,7 +216,6 @@ class Zipformer(EncoderInterface):
|
|||||||
encoder_dim[-1],
|
encoder_dim[-1],
|
||||||
downsample=output_downsampling_factor,
|
downsample=output_downsampling_factor,
|
||||||
dropout=dropout)
|
dropout=dropout)
|
||||||
self.norm = BasicNorm(num_channels=encoder_dim[-1])
|
|
||||||
|
|
||||||
|
|
||||||
def _init_skip_modules(self):
|
def _init_skip_modules(self):
|
||||||
@ -358,7 +357,6 @@ class Zipformer(EncoderInterface):
|
|||||||
lengths = (lengths + 1) // 2
|
lengths = (lengths + 1) // 2
|
||||||
|
|
||||||
x = x.permute(1, 0, 2) # (T, N, C) ->(N, T, C)
|
x = x.permute(1, 0, 2) # (T, N, C) ->(N, T, C)
|
||||||
x = self.norm(x)
|
|
||||||
|
|
||||||
return x, lengths
|
return x, lengths
|
||||||
|
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user