Remove pointless assertion

This commit is contained in:
Daniel Povey 2023-05-17 14:54:29 +08:00
parent 8fce9a05fc
commit 24e8a7a8fd

View File

@ -671,7 +671,6 @@ class SubformerEncoder(nn.Module):
if seq_len <= self.chunk_size or seq_len % self.chunk_size != 0: if seq_len <= self.chunk_size or seq_len % self.chunk_size != 0:
return [ seq_len ], [ 0 ] * len(self.layers) return [ seq_len ], [ 0 ] * len(self.layers)
else: else:
assert seq_len % self.chunk_size == 0, (seq_len, self.chunk_size)
num_layers = len(self.layers) num_layers = len(self.layers)
chunk_indexes = [0, 1] * (num_layers + 1 // 2) chunk_indexes = [0, 1] * (num_layers + 1 // 2)
return [ self.chunk_size, seq_len ], chunk_indexes[:num_layers] return [ self.chunk_size, seq_len ], chunk_indexes[:num_layers]