Kamal Raj a2dec768a2 beit-flax (#13515) ...

* beit-flax

* updated FLAX_BEIT_MLM_DOCSTRING

* removed bool_masked_pos from classification

* updated Copyright

* code refactoring: x -> embeddings

* updated test: rm from_pt

* Update docs/source/model_doc/beit.rst

* model code dtype updates and
other changes according to review

* relative_position_bias
revert back to pytorch design

2021-09-21 13:34:19 +02:00

6.5 KiB

Raw Blame History

View Raw