SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity
Files
9cf7b23b9bda5ae0e827993e8154d17065ef8dab
HuggingFace_transformer/tests
History
Julien Plu 9cf7b23b9b Custom TF weights loading (#7422) 
* First try

* Fix TF utils

* Handle authorized unexpected keys when loading weights

* Add several more authorized unexpected keys

* Apply style

* Fix test

* Address Patrick's comments.

* Update src/transformers/modeling_tf_utils.py

Co-authored-by: Sylvain Gugger <35901082+sgugger@users.noreply.github.com>

* Update src/transformers/modeling_tf_utils.py

Co-authored-by: Sylvain Gugger <35901082+sgugger@users.noreply.github.com>

* Apply style

* Make return_dict the default behavior and display a warning message

* Revert

* Replace wrong keyword

* Revert code

* Add forgot key

* Fix bug in loading PT models from a TF one.

* Fix sort

* Add a test for custom load weights in BERT

* Apply style

* Remove unused import

Co-authored-by: Sylvain Gugger <35901082+sgugger@users.noreply.github.com>
2020-10-05 09:58:45 -04:00
..
fixtures
Albert pretrain datasets/ datacollator (#6168)
2020-09-10 07:56:29 -04:00
__init__.py
GPU text generation: mMoved the encoded_prompt to correct device
2020-01-06 15:11:12 +01:00
conftest.py
ignore FutureWarning in tests (#7079)
2020-09-14 07:50:51 -04:00
test_activations_tf.py
Refactoring the TF activations functions (#7150)
2020-09-16 07:03:47 -04:00
test_activations.py
Update repo to isort v5 (#6686)
2020-08-24 11:03:01 -04:00
test_benchmark_tf.py
[Benchmarks] Change all args to from no_... to their positive form (#7075)
2020-09-23 13:25:24 -04:00
test_benchmark.py
[Benchmarks] Change all args to from no_... to their positive form (#7075)
2020-09-23 13:25:24 -04:00
test_cli.py
[transformers-cli] fix logger getter (#6777)
2020-08-27 20:01:17 -04:00
test_configuration_auto.py
Move tests/utils.py -> transformers/testing_utils.py (#5350)
2020-07-01 10:31:17 -04:00
test_configuration_common.py
Pass kwargs to configuration (#3147)
2020-03-05 17:16:57 -05:00
test_data_collator.py
Mark big downloads slow (#7325)
2020-09-22 12:21:52 -04:00
test_doc_samples.py
Move tests/utils.py -> transformers/testing_utils.py (#5350)
2020-07-01 10:31:17 -04:00
test_hf_api.py
Update repo to isort v5 (#6686)
2020-08-24 11:03:01 -04:00
test_hf_argparser.py
parse arguments from dict (#4869)
2020-07-31 04:44:23 -04:00
test_logging.py
adding TRANSFORMERS_VERBOSITY env var (#6961)
2020-09-09 04:08:01 -04:00
test_model_card.py
GPU text generation: mMoved the encoded_prompt to correct device
2020-01-06 15:11:12 +01:00
test_model_output.py
Add tests and fix various bugs in ModelOutput (#7073)
2020-09-11 12:01:33 -04:00
test_modeling_albert.py
Black 20 release
2020-08-26 17:20:22 +02:00
test_modeling_auto.py
[Longformer, Bert, Roberta, ...] Fix multi gpu training (#7272)
2020-09-25 20:33:21 +02:00
test_modeling_bart.py
[Seq2Seq] Fix a couple of bugs and clean examples (#7474)
2020-10-01 17:38:50 +02:00
test_modeling_bert_generation.py
Add "Leveraging Pretrained Checkpoints for Generation" Seq2Seq models. (#6594)
2020-09-10 16:40:51 +02:00
test_modeling_bert.py
Black 20 release
2020-08-26 17:20:22 +02:00
test_modeling_camembert.py
Black 20 release
2020-08-26 17:20:22 +02:00
test_modeling_common.py
[Seq2Seq] Fix a couple of bugs and clean examples (#7474)
2020-10-01 17:38:50 +02:00
test_modeling_ctrl.py
Black 20 release
2020-08-26 17:20:22 +02:00
test_modeling_deberta.py
Add DeBERTa model (#5929)
2020-09-30 07:07:30 -04:00
test_modeling_distilbert.py
Black 20 release
2020-08-26 17:20:22 +02:00
test_modeling_dpr.py
Black 20 release
2020-08-26 17:20:22 +02:00
test_modeling_electra.py
Black 20 release
2020-08-26 17:20:22 +02:00
test_modeling_encoder_decoder.py
clean naming (#7068)
2020-09-11 09:57:53 +02:00
test_modeling_flaubert.py
Black 20 release
2020-08-26 17:20:22 +02:00
test_modeling_fsmt.py
[Seq2Seq] Fix a couple of bugs and clean examples (#7474)
2020-10-01 17:38:50 +02:00
test_modeling_funnel.py
Fix FP16 and attention masks in FunnelTransformer (#7374)
2020-09-25 12:20:39 -04:00
test_modeling_gpt2.py
[Seq2Seq] Fix a couple of bugs and clean examples (#7474)
2020-10-01 17:38:50 +02:00
test_modeling_layoutlm.py
Add LayoutLM Model (#7064)
2020-09-22 09:28:02 -04:00
test_modeling_longformer.py
Black 20 release
2020-08-26 17:20:22 +02:00
test_modeling_lxmert.py
fix (#6946)
2020-09-04 16:08:54 +02:00
test_modeling_marian.py
Fix marian slow test (#6854)
2020-08-31 16:10:43 -04:00
test_modeling_mbart.py
modeling_bart: 3 small cleanups that dont change outputs (#7381)
2020-09-25 04:24:14 -04:00
test_modeling_mobilebert.py
Black 20 release
2020-08-26 17:20:22 +02:00
test_modeling_openai.py
[Generate] Facilitate PyTorch generate using ModelOutputs (#6735)
2020-09-01 12:38:25 +02:00
test_modeling_pegasus.py
Enable pegasus fp16 by clamping large activations (#7243)
2020-10-01 04:48:37 -04:00
test_modeling_rag.py
[RAG] Fix retrieval offset in RAG's HfIndex and better integration tests (#7372)
2020-09-25 16:12:46 +02:00
test_modeling_reformer.py
Black 20 release
2020-08-26 17:20:22 +02:00
test_modeling_roberta.py
Black 20 release
2020-08-26 17:20:22 +02:00
test_modeling_squeezebert.py
SqueezeBERT architecture (#7083)
2020-10-05 04:25:43 -04:00
test_modeling_t5.py
[Seq2Seq] Fix a couple of bugs and clean examples (#7474)
2020-10-01 17:38:50 +02:00
test_modeling_tf_albert.py
Update repo to isort v5 (#6686)
2020-08-24 11:03:01 -04:00
test_modeling_tf_auto.py
Update repo to isort v5 (#6686)
2020-08-24 11:03:01 -04:00
test_modeling_tf_bert.py
Custom TF weights loading (#7422)
2020-10-05 09:58:45 -04:00
test_modeling_tf_camembert.py
Black 20 release
2020-08-26 17:20:22 +02:00
test_modeling_tf_common.py
[Seq2Seq] Fix a couple of bugs and clean examples (#7474)
2020-10-01 17:38:50 +02:00
test_modeling_tf_ctrl.py
Black 20 release
2020-08-26 17:20:22 +02:00
test_modeling_tf_distilbert.py
test_tf_common: remove un_used mixin class parameters (#6866)
2020-09-02 10:54:40 -04:00
test_modeling_tf_electra.py
Black 20 release
2020-08-26 17:20:22 +02:00
test_modeling_tf_flaubert.py
Black 20 release
2020-08-26 17:20:22 +02:00
test_modeling_tf_funnel.py
Fix saving TF custom models (#7291)
2020-09-22 09:31:13 -04:00
test_modeling_tf_gpt2.py
[Seq2Seq] Fix a couple of bugs and clean examples (#7474)
2020-10-01 17:38:50 +02:00
test_modeling_tf_longformer.py
test_tf_common: remove un_used mixin class parameters (#6866)
2020-09-02 10:54:40 -04:00
test_modeling_tf_lxmert.py
Adding the LXMERT pretraining model (MultiModal languageXvision) to HuggingFace's suite of models (#5793)
2020-09-03 04:02:25 -04:00
test_modeling_tf_mobilebert.py
Update repo to isort v5 (#6686)
2020-08-24 11:03:01 -04:00
test_modeling_tf_openai.py
[Generate] Facilitate PyTorch generate using ModelOutputs (#6735)
2020-09-01 12:38:25 +02:00
test_modeling_tf_roberta.py
Black 20 release
2020-08-26 17:20:22 +02:00
test_modeling_tf_t5.py
[Seq2Seq] Fix a couple of bugs and clean examples (#7474)
2020-10-01 17:38:50 +02:00
test_modeling_tf_transfo_xl.py
test_tf_common: remove un_used mixin class parameters (#6866)
2020-09-02 10:54:40 -04:00
test_modeling_tf_xlm_roberta.py
Update repo to isort v5 (#6686)
2020-08-24 11:03:01 -04:00
test_modeling_tf_xlm.py
Black 20 release
2020-08-26 17:20:22 +02:00
test_modeling_tf_xlnet.py
test_tf_common: remove un_used mixin class parameters (#6866)
2020-09-02 10:54:40 -04:00
test_modeling_transfo_xl.py
Black 20 release
2020-08-26 17:20:22 +02:00
test_modeling_xlm_roberta.py
Update repo to isort v5 (#6686)
2020-08-24 11:03:01 -04:00
test_modeling_xlm.py
Black 20 release
2020-08-26 17:20:22 +02:00
test_modeling_xlnet.py
[Seq2Seq] Fix a couple of bugs and clean examples (#7474)
2020-10-01 17:38:50 +02:00
test_onnx.py
Fix flaky ONNX tests (#6531)
2020-08-17 09:04:35 -04:00
test_optimization_tf.py
Update repo to isort v5 (#6686)
2020-08-24 11:03:01 -04:00
test_optimization.py
Format
2020-08-27 18:31:51 +02:00
test_pipelines.py
Mark big downloads slow (#7325)
2020-09-22 12:21:52 -04:00
test_retrieval_rag.py
fix rag retriever save pretrained (#7399)
2020-09-25 19:47:12 +02:00
test_skip_decorators.py
[testing] skip decorators: docs, tests, bugs (#7334)
2020-09-23 05:16:19 -04:00
test_tokenization_albert.py
[HUGE] Refactoring tokenizers backend - padding - truncation - pre-tokenized pipeline - fast tokenizers - tests (#4510)
2020-06-15 17:12:51 -04:00
test_tokenization_auto.py
[from_pretrained] Allow tokenizer_type ≠ model_type (#6995)
2020-09-09 04:22:59 -04:00
test_tokenization_bart.py
[tests] fix typos in inputs (#6818)
2020-08-30 18:19:57 +08:00
test_tokenization_bert_generation.py
Check decorator order (#7326)
2020-09-24 04:54:37 -04:00
test_tokenization_bert_japanese.py
Support additional dictionaries for BERT Japanese tokenizers (#6515)
2020-08-17 12:00:23 +08:00
test_tokenization_bert.py
Add strip_accents to basic BertTokenizer. (#6280)
2020-08-06 18:52:28 +08:00
test_tokenization_bertweet.py
Add new pre-trained models BERTweet and PhoBERT (#6129)
2020-09-18 13:16:43 -04:00
test_tokenization_common.py
Check decorator order (#7326)
2020-09-24 04:54:37 -04:00
test_tokenization_ctrl.py
[HUGE] Refactoring tokenizers backend - padding - truncation - pre-tokenized pipeline - fast tokenizers - tests (#4510)
2020-06-15 17:12:51 -04:00
test_tokenization_deberta.py
Add DeBERTa model (#5929)
2020-09-30 07:07:30 -04:00
test_tokenization_distilbert.py
Move tests/utils.py -> transformers/testing_utils.py (#5350)
2020-07-01 10:31:17 -04:00
test_tokenization_dpr.py
Fix tests imports dpr (#5576)
2020-07-07 16:35:12 +02:00
test_tokenization_fast.py
is_pretokenized -> is_split_into_words (#7236)
2020-09-22 09:34:35 -04:00
test_tokenization_fsmt.py
[ported model] FSMT (FairSeq MachineTranslation) (#6940)
2020-09-17 11:31:29 -04:00
test_tokenization_funnel.py
Funnel transformer (#6908)
2020-09-08 08:08:08 -04:00
test_tokenization_gpt2.py
[HUGE] Refactoring tokenizers backend - padding - truncation - pre-tokenized pipeline - fast tokenizers - tests (#4510)
2020-06-15 17:12:51 -04:00
test_tokenization_layoutlm.py
Add LayoutLM Model (#7064)
2020-09-22 09:28:02 -04:00
test_tokenization_lxmert.py
Adding the LXMERT pretraining model (MultiModal languageXvision) to HuggingFace's suite of models (#5793)
2020-09-03 04:02:25 -04:00
test_tokenization_marian.py
rename prepare_translation_batch -> prepare_seq2seq_batch (#6103)
2020-08-11 15:57:07 -04:00
test_tokenization_mbart.py
prepare_seq2seq_batch makes labels/ decoder_input_ids made later. (#6654)
2020-08-28 11:15:17 -04:00
test_tokenization_openai.py
[HUGE] Refactoring tokenizers backend - padding - truncation - pre-tokenized pipeline - fast tokenizers - tests (#4510)
2020-06-15 17:12:51 -04:00
test_tokenization_pegasus.py
prepare_seq2seq_batch makes labels/ decoder_input_ids made later. (#6654)
2020-08-28 11:15:17 -04:00
test_tokenization_phobert.py
Add new pre-trained models BERTweet and PhoBERT (#6129)
2020-09-18 13:16:43 -04:00
test_tokenization_rag.py
RAG (#6813)
2020-09-22 18:29:58 +02:00
test_tokenization_reformer.py
Check decorator order (#7326)
2020-09-24 04:54:37 -04:00
test_tokenization_roberta.py
prepare_seq2seq_batch makes labels/ decoder_input_ids made later. (#6654)
2020-08-28 11:15:17 -04:00
test_tokenization_squeezebert.py
SqueezeBERT architecture (#7083)
2020-10-05 04:25:43 -04:00
test_tokenization_t5.py
[T5Tokenizer] remove prefix_tokens (#7078)
2020-09-11 14:18:45 -04:00
test_tokenization_transfo_xl.py
Transformer-XL: Improved tokenization with sacremoses (#6322)
2020-08-28 09:56:17 -04:00
test_tokenization_utils.py
Fixes to make life easier with the nlp library (#6423)
2020-08-12 08:00:56 -04:00
test_tokenization_xlm_roberta.py
Move tests/utils.py -> transformers/testing_utils.py (#5350)
2020-07-01 10:31:17 -04:00
test_tokenization_xlm.py
Move tests/utils.py -> transformers/testing_utils.py (#5350)
2020-07-01 10:31:17 -04:00
test_tokenization_xlnet.py
Move tests/utils.py -> transformers/testing_utils.py (#5350)
2020-07-01 10:31:17 -04:00
test_trainer_distributed.py
Add tests to Trainer (#6605)
2020-08-20 11:13:50 -04:00
test_trainer.py
Expand test to locate flakiness (#7580)
2020-10-05 09:45:47 -04:00
test_utils_check_copies.py
Get a better error when check_copies fails (#7457)
2020-09-30 10:05:14 +02:00