Skip to content

Commit 0a1647a

Browse files
discortvincentqb
authored andcommitted
fixed tests for deepspeech
1 parent bb7432f commit 0a1647a

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

test/torchaudio_unittest/models_test.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -190,4 +190,4 @@ def test_deepspeech(self):
190190
x = torch.rand(batch_size, num_channels, input_length, num_features)
191191
out = model(x)
192192

193-
assert out.size() == (input_length, batch_size, num_classes)
193+
assert out.size() == (batch_size, input_length, num_classes)

0 commit comments

Comments
 (0)