transformers/tests/data2vec
Patrick von Platen 8d3f952adb
[Data2Vec] Add data2vec vision (#16760)
* save intermediate

* add vision

* add vision

* save

* finish models

* finish models

* continue

* finish

* up

* up

* up

* tests all pass

* clean up

* up

* up

* fix bugs in beit

* correct docs

* finish

* finish docs

* make style

* up

* more fixes

* fix type hint

* make style

* Apply suggestions from code review

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

* Update tests/data2vec/test_modeling_data2vec_vision.py

Co-authored-by: NielsRogge <48327001+NielsRogge@users.noreply.github.com>

* fix test

Co-authored-by: NielsRogge <48327001+NielsRogge@users.noreply.github.com>
Co-authored-by: Sylvain Gugger <35901082+sgugger@users.noreply.github.com>
2022-04-18 17:52:13 +02:00
..
__init__.py Add Data2Vec (#15507) 2022-03-01 11:09:20 +01:00
test_modeling_data2vec_audio.py Enable more test_torchscript (#16679) 2022-04-11 18:23:35 +02:00
test_modeling_data2vec_text.py Add Data2Vec (#15507) 2022-03-01 11:09:20 +01:00
test_modeling_data2vec_vision.py [Data2Vec] Add data2vec vision (#16760) 2022-04-18 17:52:13 +02:00