Phil Wang 6db20debb4 add patch merger

2022-03-01 16:50:17 -08:00

..

__init__.py

release masked autoencoder

2021-11-12 20:08:48 -08:00

ats_vit.py

add adaptive token sampling paper

2021-12-03 19:52:40 -08:00

cait.py

add layer dropout for CaiT

2021-04-01 20:30:37 -07:00

cct.py

Minor changes

2021-07-01 17:51:35 -07:00

cross_vit.py

add documentation for three recent vision transformer follow-up papers

2021-03-31 09:22:15 -07:00

crossformer.py

epsilon should be inside square root

2022-01-24 17:24:41 -08:00

cvt.py

epsilon should be inside square root

2022-01-24 17:24:41 -08:00

deepvit.py

add deep vit

2021-03-23 11:57:13 -07:00

dino.py

implement SOTA new self-supervised learning technique from facebook for vision transformers, Dino

2021-05-02 14:00:36 -07:00

distill.py

fix hard distillation, thanks to @CiaoHe

2021-08-12 08:40:57 -07:00

efficient.py

allow for rectangular images for efficient adapter

2022-01-31 08:55:31 -08:00

extractor.py

add ViT for small datasets https://arxiv.org/abs/2112.13492

2021-12-28 10:58:21 -08:00

levit.py

use hardswish for levit

2021-08-19 08:22:55 -07:00

local_vit.py

fix pooling bugs across a few new archs

2021-04-19 22:36:23 -07:00

mae.py

decoder positional embedding needs to be reapplied https://twitter.com/giffmana/status/1479195631587631104

2022-01-06 13:14:41 -08:00

mobile_vit.py

Update MobileViT

2021-12-30 05:52:23 +08:00

mpp.py

fix mpp

2021-06-16 16:46:32 -07:00

nest.py

epsilon should be inside square root

2022-01-24 17:24:41 -08:00

pit.py

make initial channels customizable for PiT

2021-11-22 18:08:49 -08:00

recorder.py

return None from extractor if no attention layers

2022-01-28 17:49:58 -08:00

regionvit.py

cleanup region vit

2021-11-22 12:42:24 -08:00

rvt.py

move freqs in RvT to linspace

2021-10-05 09:23:44 -07:00

simmim.py

add SimMIM

2021-11-21 15:50:19 -08:00

t2t.py

Make model process any scale image

2021-08-21 22:35:26 +09:00

twins_svt.py

epsilon should be inside square root

2022-01-24 17:24:41 -08:00

vit_for_small_dataset.py

add ViT for small datasets https://arxiv.org/abs/2112.13492

2021-12-28 10:58:21 -08:00

vit_with_patch_merger.py

add patch merger

2022-03-01 16:50:17 -08:00

vit.py

remove unused package

2021-08-30 18:25:03 +09:00