transformers/tests/realm
Li-Huai (Allan) Lin 7b3bd1f21a
Fix and improve REALM fine-tuning (#15297)
* Draft

* Add test

* Update src/transformers/models/realm/modeling_realm.py

* Apply suggestion

* Add block_mask

* Update

* Update

* Add block_embedding_to

* Remove no_grad

* Use AutoTokenizer

* Remove model.to overridding
2022-03-03 14:10:15 +01:00
..
__init__.py [Test refactor 1/5] Per-folder tests reorganization (#15725) 2022-02-23 15:46:28 -05:00
test_modeling_realm.py Fix and improve REALM fine-tuning (#15297) 2022-03-03 14:10:15 +01:00
test_retrieval_realm.py Fix and improve REALM fine-tuning (#15297) 2022-03-03 14:10:15 +01:00
test_tokenization_realm.py [Test refactor 1/5] Per-folder tests reorganization (#15725) 2022-02-23 15:46:28 -05:00