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