Cosmetic fix to formula

This commit is contained in:
Daniel Povey 2023-02-11 15:38:03 +08:00
parent 3cb43c3c36
commit 49627c6251

View File

@ -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:]