add back fairseq manual_seed

add back rnnt_test
pytorch · Jun 2, 2022 · 325653b · 325653b
1 parent e410998
commit 325653b
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/test/torchaudio_unittest/models/wav2vec2/fairseq_integration_test.py b/test/torchaudio_unittest/models/wav2vec2/fairseq_integration_test.py
@@ -205,6 +205,7 @@ def test_import_finetuning_model(self, config, _):
         imported = import_fairseq_model(original).eval()
 
         # Without mask
+        torch.manual_seed(0)
         x = torch.randn(batch_size, num_frames)
         ref = original(x, torch.zeros_like(x))["encoder_out"].transpose(0, 1)
         hyp, _ = imported(x)

diff --git a/test/torchaudio_unittest/prototype/rnnt_test_impl.py b/test/torchaudio_unittest/prototype/rnnt_test_impl.py
@@ -84,6 +84,7 @@ def _get_joiner_input(self):
 
     def setUp(self):
         super().setUp()
+        torch.random.manual_seed(31)
 
     def test_torchscript_consistency_forward(self):
         r"""Verify that scripting RNNT does not change the behavior of method `forward`."""