More benchmarks

This commit is contained in:
Michael Hansen
2023-04-10 14:37:34 -05:00
parent 082e64a53c
commit c945c141f1
4 changed files with 120 additions and 12 deletions

View File

@@ -57,9 +57,7 @@ def main():
)
sequence_lengths = torch.LongTensor([sequences.size(1)])
sid: Optional[int] = None
if num_speakers > 1:
sid = torch.LongTensor([0])
sid = torch.LongTensor([0])
dummy_input = (
sequences,