Set seed for head_masking test

2019-08-30 23:58:49 -04:00
parent 0fd0b674e6
commit 7f006cdd87
1 changed files with 1 additions and 0 deletions
--- 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