mmpretrain/mmcls/models/backbones
Ma Zerun fffa30dd48
[Feature] Add Tokens-to-Token ViT backbone and converted checkpoints. (#467)
* add t2t backbone

* register t2t_vit

* add t2t_vit config

* [Temp] Align posterize transform with timm.

* Fix lint

* Refactor t2t-vit

* Add config for t2t-vit

* Add metafile and README for t2t-vit

* Add unit tests

* configs

* Update metafile and README

* Improve docstring

* Fix batch size which should be 8x64 instead of 8x128

* Fix typo

* Update model zoo

* Update training augments config.

* Move some arguments of T2TModule to T2TViT

* Update docs.

* Update unit test

Co-authored-by: HIT-cwh <2892770585@qq.com>
2021-10-29 10:37:16 +08:00
..
__init__.py [Feature] Add Tokens-to-Token ViT backbone and converted checkpoints. (#467) 2021-10-29 10:37:16 +08:00
alexnet.py [Enhance] Improve downstream repositories compatibility (#421) 2021-09-08 10:38:57 +08:00
base_backbone.py [Docs] Add Copyright information. (#413) 2021-08-17 19:52:42 +08:00
lenet.py [Enhance] Improve downstream repositories compatibility (#421) 2021-09-08 10:38:57 +08:00
mobilenet_v2.py [Enhance] Improve downstream repositories compatibility (#421) 2021-09-08 10:38:57 +08:00
mobilenet_v3.py [Docs] Add code-spell pre-commit hook and fix a large mount of typos. (#470) 2021-10-13 14:33:07 +08:00
regnet.py [Fix] Fix missing import `Compose` and `Normalize`. 2021-10-28 15:21:05 +08:00
repvgg.py [Docs] Add code-spell pre-commit hook and fix a large mount of typos. (#470) 2021-10-13 14:33:07 +08:00
res2net.py [Feature] Add Res2Net backbone and converted weights. (#465) 2021-10-20 16:34:22 +08:00
resnest.py [Docs] Add Copyright information. (#413) 2021-08-17 19:52:42 +08:00
resnet.py [Feature] Add Res2Net backbone and converted weights. (#465) 2021-10-20 16:34:22 +08:00
resnet_cifar.py [Enhance] Improve downstream repositories compatibility (#421) 2021-09-08 10:38:57 +08:00
resnext.py [Docs] Add Copyright information. (#413) 2021-08-17 19:52:42 +08:00
seresnet.py [Docs] Add Copyright information. (#413) 2021-08-17 19:52:42 +08:00
seresnext.py [Docs] Add Copyright information. (#413) 2021-08-17 19:52:42 +08:00
shufflenet_v1.py [Enhance] Improve downstream repositories compatibility (#421) 2021-09-08 10:38:57 +08:00
shufflenet_v2.py [Enhance] Improve downstream repositories compatibility (#421) 2021-09-08 10:38:57 +08:00
swin_transformer.py [Enhance] Improve downstream repositories compatibility (#421) 2021-09-08 10:38:57 +08:00
t2t_vit.py [Feature] Add Tokens-to-Token ViT backbone and converted checkpoints. (#467) 2021-10-29 10:37:16 +08:00
timm_backbone.py [Enhance] Improve downstream repositories compatibility (#421) 2021-09-08 10:38:57 +08:00
tnt.py [Docs] Add code-spell pre-commit hook and fix a large mount of typos. (#470) 2021-10-13 14:33:07 +08:00
vgg.py [Docs] Add code-spell pre-commit hook and fix a large mount of typos. (#470) 2021-10-13 14:33:07 +08:00
vision_transformer.py [Refactor] Refator ViT (Continue #295) (#395) 2021-10-18 16:07:00 +08:00