Set seed for head_masking test

7f006cdd · LysandreJik · 0fd0b674 · 7f006cdd
Commit 7f006cdd authored Aug 30, 2019 by LysandreJik
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

pytorch_transformers/tests/modeling_common_test.py pytorch_transformers/tests/modeling_common_test.py +1 -0

No files found.
--- a/pytorch_transformers/tests/modeling_common_test.py
+++ b/pytorch_transformers/tests/modeling_common_test.py
@@ -163,6 +163,7 @@ class CommonTestCases:
            if not self.test_head_masking:
                return

+            torch.manual_seed(42)
            config, inputs_dict = self.model_tester.prepare_config_and_inputs_for_common()

            config.output_attentions = True