PR #17354 add MobileViT model

hollance added New model

hollance requested a review from

NielsRogge 3 years ago

NielsRogge commented on 2022-05-19

hollance marked this pull request as ready for review 3 years ago

hollance force pushed 3 years ago

hollance requested a review from

NielsRogge 3 years ago

sgugger approved these changes on 2022-05-31

hollance force pushed 3 years ago

NielsRogge commented on 2022-06-01

NielsRogge commented on 2022-06-02

hollance force pushed 3 years ago

hollance requested a review from

NielsRogge 3 years ago

hollance force pushed 3 years ago

add MobileViT

17ae4fbd

fixup

e2d7775b

Update README.md

0809213d

remove empty line

cd7fd8a3

use clearer variable names

10fa2814

rename to MobileViTTransformerLayer

61a3f869

no longer inherit from nn.Sequential

eaaa6965

fixup

1d3ea33f

fixup

acc6c64a

not sure why this got added twice

a5885231

rename organization for checkpoints

3b66e071

fix it up

7f2e365f

Update src/transformers/models/mobilevit/__init__.py

54911d09

Update src/transformers/models/mobilevit/configuration_mobilevit.py

2c6b0f2b

Update src/transformers/models/mobilevit/configuration_mobilevit.py

f9248c25

Update src/transformers/models/mobilevit/configuration_mobilevit.py

21fc5c9b

Update tests/models/mobilevit/test_modeling_mobilevit.py

d12013e6

Update src/transformers/models/mobilevit/modeling_mobilevit.py

bf1acf2b

Update src/transformers/models/mobilevit/modeling_mobilevit.py

f439a554

Update src/transformers/models/mobilevit/modeling_mobilevit.py

59c5043c

Update src/transformers/models/mobilevit/modeling_mobilevit.py

a784c9ea

code style improvements

990f66ac

fixup

c4085617

Update docs/source/en/model_doc/mobilevit.mdx

c8769c0c

Update docs/source/en/model_doc/mobilevit.mdx

c7afd1af

Update src/transformers/models/mobilevit/configuration_mobilevit.py

7a54f196

Update src/transformers/models/mobilevit/configuration_mobilevit.py

bf743ce3

download labels from hub

68b898e5

rename layers

dfdede5e

rename more layers

ea6e5a42

don't compute loss in separate function

1bcfca6d

remove some nn.Sequential

08a33bea

replace nn.Sequential with new MobileViTTransformer class

72d28643

replace nn.Sequential with MobileViTMobileNetLayer

22460e1c

fix pruning since model structure changed

10d30bbe

fixup

49a69ae9

fix doc comment

87ed8fd2

remove custom resize from feature extractor

e904cc48

fix ONNX import

25fcc7b0

add to doc tests

84e29340

use center_crop from image_utils

49ea03e6

move RGB->BGR flipping into image_utils

9e30ff39

fix broken tests

414b5025

wrong type hint

25c1b98d

small tweaks

a0ec9d5d

hollance force pushed to a0ec9d5d 3 years ago

NielsRogge commented on 2022-06-28

sgugger merged fbc7598b into main 3 years ago

transformers
add MobileViT model
#17354

Merged

add MobileViT model #17354

transformers add MobileViT model #17354 Merged

add MobileViT model #17354

transformers
add MobileViT model
#17354

Merged