SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Set seed for head_masking test

Browse Source
This commit is contained in:
LysandreJik
2019-08-30 23:58:49 -04:00
parent 0fd0b674e6
commit 7f006cdd87
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
pytorch_transformers/tests/modeling_common_test.py
View File

@@ -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