Skip to content

Commit

Permalink
fix conformer test
Browse files Browse the repository at this point in the history
Signed-off-by: WoodieDudy <goshagks@gmail.com>
  • Loading branch information
WoodieDudy committed Aug 21, 2024
1 parent 6a32a7a commit d282a59
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion tests/collections/asr/test_conformer_encoder.py
Original file line number Diff line number Diff line change
Expand Up @@ -81,7 +81,7 @@ def test_stochastic_depth_model_creation(self):
def test_stochastic_depth_forward(self):
"""Testing that forward works and we get randomness during training, but not during eval."""
random_input = torch.rand((1, 2, 2))
random_length = torch.tensor([2, 2], dtype=torch.int64)
random_length = torch.tensor([2], dtype=torch.int64)

model = ConformerEncoder(
feat_in=2,
Expand Down

0 comments on commit d282a59

Please sign in to comment.