Skip to content

GitLab

Commit 7f006cdd authored by LysandreJik's avatar LysandreJik
Browse files

Set seed for head_masking test

parent 0fd0b674
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
pytorch_transformers/tests/modeling_common_test.py
View file @ 7f006cdd
...@@ -163,6 +163,7 @@ class CommonTestCases: ...@@ -163,6 +163,7 @@ class CommonTestCases:
if not self.test_head_masking: if not self.test_head_masking:
return return
torch.manual_seed(42)
config, inputs_dict = self.model_tester.prepare_config_and_inputs_for_common() config, inputs_dict = self.model_tester.prepare_config_and_inputs_for_common()
config.output_attentions = True config.output_attentions = True
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment