NielsRogge/transformers

Pull Requests Commits

Address additional comments

NielsRogge committed 4 years ago

c6c0f272

NielsRogge committed 4 years ago

37ae119a

Add ViTFeatureExtractor to conversion script

NielsRogge committed 4 years ago

8b95a1eb

Add pooler option to ViTForImageClassification, improve docs

NielsRogge committed 4 years ago

e790c1d2

Rename self.self to self.attention

NielsRogge committed 4 years ago

472f96d7

Update conversion script

NielsRogge committed 4 years ago

f9a1ac6d

Address most comments by @sgugger @LysandreJik

NielsRogge committed 4 years ago

f35360e9

Revert previous change

NielsRogge committed 4 years ago

884b7a7d

One more bug fix

NielsRogge committed 4 years ago

f6556b5f

Fix bugs in tests

NielsRogge committed 4 years ago

03b76389

Change normalization to match original implementation

NielsRogge committed 4 years ago

852b7771

Some docs improvements + clearer input checking for ViTFeatureExtractor

NielsRogge committed 4 years ago

f5ba2f4b

Merge branch 'master' into modeling_vit_pytorch_v2

NielsRogge committed 4 years ago

Verified 02c06bcd

Remove attention mask, add support for head mask

NielsRogge committed 4 years ago

0e02f645

Sort init import (#10801)

sgugger committed 4 years ago

Verified 21e86f99

Make style & quality, docs improvements

NielsRogge committed 4 years ago

e01294c6

All tests are passing

NielsRogge committed 4 years ago

647f0e40

NielsRogge committed 4 years ago

466cef1a

NielsRogge committed 4 years ago

a7a9e0e1

24 model tests pass, 6 fail on cpu

NielsRogge committed 4 years ago

872ae16d

Improve model tests

NielsRogge committed 4 years ago

9637b856

Make all tests of ViTFeatureExtractor pass

NielsRogge committed 4 years ago

5cd7dfdf

Use append instead of extend

NielsRogge committed 4 years ago

c3dfbe6a

Fix another bug with init

NielsRogge committed 4 years ago

4c91fb32

Fix import of ViTFeatureExtractor

NielsRogge committed 4 years ago

4fb8def7

Add is_torchvision_available to general init of vit

NielsRogge committed 4 years ago

43ba11f9

Improve model tests

NielsRogge committed 4 years ago

b2b34328

Remove test_image_processor_common

NielsRogge committed 4 years ago

20e3d1eb

Add documentation

NielsRogge committed 4 years ago

c39f1559

Add torch.stack

NielsRogge committed 4 years ago

b168ee48

Older