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

Output attention takes an s (#6903)

Browse Source 
* Fix output_attention -> output_attentions

* Formatting

* One unsaved file
This commit is contained in:
Sylvain Gugger
2020-09-02 08:11:45 -04:00
committed by GitHub
parent 485da7222f
commit 8f2723caf0
11 changed files with 75 additions and 63 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
src/transformers/configuration_utils.py
View File

@@ -300,11 +300,11 @@ class PretrainedConfig(object):
config = BertConfig.from_pretrained('bert-base-uncased') # Download configuration from S3 and cache.
config = BertConfig.from_pretrained('./test/saved_model/') # E.g. config (or model) was saved using `save_pretrained('./test/saved_model/')`
config = BertConfig.from_pretrained('./test/saved_model/my_configuration.json')
config = BertConfig.from_pretrained('bert-base-uncased', output_attention=True, foo=False)
assert config.output_attention == True
config, unused_kwargs = BertConfig.from_pretrained('bert-base-uncased', output_attention=True,
config = BertConfig.from_pretrained('bert-base-uncased', output_attentions=True, foo=False)
assert config.output_attentions == True
config, unused_kwargs = BertConfig.from_pretrained('bert-base-uncased', output_attentions=True,
foo=False, return_unused_kwargs=True)
assert config.output_attention == True
assert config.output_attentions == True
assert unused_kwargs == {'foo': False}
"""