mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-08-26 18:24:18 +00:00
Fix comments
This commit is contained in:
parent
06a86f50b9
commit
5bfcf65cca
@ -188,7 +188,7 @@ class Transformer(nn.Module):
|
|||||||
x = self.feat_batchnorm(x)
|
x = self.feat_batchnorm(x)
|
||||||
x = x.permute(0, 2, 1) # (N, C, T) -> (N, T, C)
|
x = x.permute(0, 2, 1) # (N, C, T) -> (N, T, C)
|
||||||
if isinstance(self.use_feat_batchnorm, float):
|
if isinstance(self.use_feat_batchnorm, float):
|
||||||
x *= 0.1
|
x *= self.use_feat_batchnorm
|
||||||
encoder_memory, memory_key_padding_mask = self.run_encoder(
|
encoder_memory, memory_key_padding_mask = self.run_encoder(
|
||||||
x, supervision
|
x, supervision
|
||||||
)
|
)
|
||||||
|
Loading…
x
Reference in New Issue
Block a user