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

Model Templates for Seq2Seq (#9251)

Browse Source 
* adapt cookie cutter

* fix copy past statement

* delete copy statements for now

* remove unused import from template

* make doc rst

* correct config docstring

* correct training

* correct inputs processing tf enc dec

* make style

* adapt templates

* clean tabs

* correct tensor -> Tensor naming

* correct indent

* correct templates

* fix the test

* break lines to avoid > 119

* Apply suggestions from code review
This commit is contained in:
Patrick von Platen
2020-12-22 23:41:20 +01:00
committed by GitHub
parent e6c1f1cad8
commit cbe63949d7
21 changed files with 3561 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
templates/adding_a_new_model/cookiecutter.json
View File

@@ -5,6 +5,7 @@
"camelcase_modelname": "BrandNewBert",
"authors": "The HuggingFace Team",
"checkpoint_identifier": "brand-new-bert-base-cased",
"tokenizer_type": ["Based on BERT", "Standalone"],
"generate_tensorflow_and_pytorch": ["PyTorch & TensorFlow", "PyTorch", "TensorFlow"]
}
"tokenizer_type": ["Based on BERT", "Based on BART", "Standalone"],
"generate_tensorflow_and_pytorch": ["PyTorch & TensorFlow", "PyTorch", "TensorFlow"],
"is_encoder_decoder_model": ["True", "False"]
}