mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-12-11 06:55:27 +00:00
Cosmetic fix to formula
This commit is contained in:
parent
3cb43c3c36
commit
49627c6251
@ -1123,7 +1123,7 @@ class ChunkCausalDepthwiseConv1d(torch.nn.Module):
|
|||||||
|
|
||||||
x = torch.nn.functional.pad(x, (left_pad, right_pad))
|
x = torch.nn.functional.pad(x, (left_pad, right_pad))
|
||||||
|
|
||||||
x_causal = self.causal_conv(x[..., :seq_len + left_pad])
|
x_causal = self.causal_conv(x[..., :left_pad + seq_len])
|
||||||
assert x_causal.shape == (batch_size, num_channels, seq_len)
|
assert x_causal.shape == (batch_size, num_channels, seq_len)
|
||||||
|
|
||||||
x_chunk = x[..., left_pad:]
|
x_chunk = x[..., left_pad:]
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user