mirrors
/
mmclassification
mirror of https://github.com/open-mmlab/mmclassification.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
mmclassification/mmcls/models/backbones
History
whcao 359f56ad58
[Feature] Add transformer in transformer (#339) 
* add tnt_small configs

* add tnt backbone

* test tnt

* add tnt to model_zoo

* rename the config file name

* add optimizor

* move tnt backbone unitest

* add metric

* fix keyname in arch

* encapsulate "inner transformer block" and "outer transformer block"

* fix TnT

* Use `inner_block_cfg` and `outer_block_cfg` instead of `args` and
`kwargs`.

Co-authored-by: mzr1996 <mzr1996@163.com>
 		2021-07-30 09:16:27 +08:00
..
__init__.py [Feature] Add transformer in transformer (#339) 2021-07-30 09:16:27 +08:00
alexnet.py revise docstr of alexnet to correct input_size (#26) 2020-08-12 16:14:19 +08:00
base_backbone.py [Feature]Delete comments (#298) 2021-06-12 21:45:22 +08:00
lenet.py
…
mobilenet_v2.py [Fix] Fix init_weights bug in some backbones and update Readme (#318) 2021-06-30 19:41:58 +08:00
mobilenet_v3.py Fix Mobilenetv3 structure and add pretrained model (#291) 2021-06-27 23:19:36 +08:00
regnet.py [Docs] Add API Reference in the docs (#342) 2021-07-14 15:06:50 +08:00
resnest.py [Docs] Add API Reference in the docs (#342) 2021-07-14 15:06:50 +08:00
resnet.py [Docs] Add API Reference in the docs (#342) 2021-07-14 15:06:50 +08:00
resnet_cifar.py
…
resnext.py [Docs] Add API Reference in the docs (#342) 2021-07-14 15:06:50 +08:00
seresnet.py [Docs] Add API Reference in the docs (#342) 2021-07-14 15:06:50 +08:00
seresnext.py [Docs] Add API Reference in the docs (#342) 2021-07-14 15:06:50 +08:00
shufflenet_v1.py [Fix] Fix docstring typo and init error in ShuffleNetV1 (#374) 2021-07-27 19:17:42 +08:00
shufflenet_v2.py [Fix] Fix init_weights bug in some backbones and update Readme (#318) 2021-06-30 19:41:58 +08:00
swin_transformer.py [Fix] Fix `patch_cfg` argument bug in SwinTransformer (#368) 2021-07-27 19:19:44 +08:00
tnt.py [Feature] Add transformer in transformer (#339) 2021-07-30 09:16:27 +08:00
vgg.py [Feature]Delete comments (#298) 2021-06-12 21:45:22 +08:00
vision_transformer.py Fix duplicate `init_weights` call in ViT init function. (#373) 2021-07-26 05:33:11 -04:00