transformers/utils
NielsRogge 1ac698744c
Add YOLOS (#16848)
* First draft

* Add YolosForObjectDetection

* Make forward pass work

* Add mid position embeddings

* Add interpolation of position encodings

* Add expected values

* Add YOLOS to tests

* Add integration test

* Support tiny model as well

* Support all models in conversion script

* Remove mid_pe_size attribute

* Make more tests pass

* Add model to README and fix config

* Add copied from statements

* Rename base_model_prefix to vit

* Add missing YOLOS_PRETRAINED_CONFIG_ARCHIVE_MAP

* Apply suggestions from code review

* Apply more suggestions from code review

* Convert remaining checkpoints

* Improve docstrings

* Add YolosFeatureExtractor

* Add feature extractor to docs

* Add corresponding tests

* Fix style

* Fix docs

* Apply suggestion from code review

* Fix bad rebase

* Fix some more bad rebase

* Fix missing character

* Improve docs and variable names

Co-authored-by: Niels Rogge <nielsrogge@Nielss-MacBook-Pro.local>
2022-05-02 18:30:55 +02:00
..
test_module Fix from_pretrained with default base_model_prefix (#15814) 2022-02-24 11:43:51 +01:00
tf_ops Check TF ops for ONNX compliance (#10025) 2021-02-15 07:55:10 -05:00
check_config_docstrings.py Add a check on config classes docstring checkpoints (#17012) 2022-04-30 10:40:46 +02:00
check_copies.py Enable doc in Spanish (#16518) 2022-04-04 10:25:46 -04:00
check_dummies.py Reorganize file utils (#16264) 2022-03-23 10:26:33 -04:00
check_inits.py Ignore empty subfolders when identifying submodules (#15204) 2022-01-18 09:48:46 -05:00
check_repo.py Update check_models_are_tested to deal with Windows path (#16973) 2022-04-28 15:31:57 +02:00
check_table.py Enable doc in Spanish (#16518) 2022-04-04 10:25:46 -04:00
check_tf_ops.py Check TF ops for ONNX compliance (#10025) 2021-02-15 07:55:10 -05:00
custom_init_isort.py Fix custom init sorting script (#16864) 2022-04-20 17:05:39 -04:00
documentation_tests.txt Add YOLOS (#16848) 2022-05-02 18:30:55 +02:00
download_glue_data.py Raise exceptions instead of asserts (#13907) 2021-10-07 12:44:23 +05:30
get_modified_files.py Updates the default branch from master to main (#16326) 2022-03-23 03:46:59 -04:00
notification_service_deprecated.py Fix the push run (#15807) 2022-02-24 19:30:17 +01:00
notification_service_doc_tests.py Add Slack notification support for doc tests (#16253) 2022-03-21 11:33:18 +01:00
notification_service.py Framework split model report (#15825) 2022-02-25 12:00:00 -05:00
prepare_for_doc_test.py [DocTests Speech] Add doc tests for all speech models (#15031) 2022-01-27 14:29:31 +01:00
print_env_pt.py [CI] move env print to util, add pt, nccl versions (#14607) 2021-12-03 08:18:36 -05:00
release.py Rename master to main for notebooks links and leftovers (#16397) 2022-03-25 09:12:23 -04:00
tests_fetcher.py Add tests for no_trainer and fix existing examples (#16656) 2022-04-08 10:03:56 -04:00
update_metadata.py Replace commit sha by commit url for update jobs (#14852) 2021-12-21 11:17:11 -05:00