SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

[Data2Vec] Add data2vec vision (#16760)

Browse Source 
* 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>
This commit is contained in:
Patrick von Platen
2022-04-18 17:52:13 +02:00
committed by GitHub
parent 33cd4be576
commit 8d3f952adb
15 changed files with 2362 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
docs/source/en/serialization.mdx
View File

@@ -54,6 +54,7 @@ Ready-made configurations include the following architectures:
- BlenderbotSmall
- CamemBERT
- Data2VecText
- Data2VecVision
- DistilBERT
- ELECTRA
- FlauBERT