mmpretrain/mmcls/models
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
..
backbones [Feature] Add transformer in transformer (#339) 2021-07-30 09:16:27 +08:00
classifiers [Docs] Add API Reference in the docs (#342) 2021-07-14 15:06:50 +08:00
heads Add `is_tracing` helper function to fix a tracing bug in PyTorch 1.6 (#347) 2021-07-07 11:55:53 +08:00
losses [Fix] Use zero as default value of `thrs` in metrics. (#341) 2021-07-18 16:57:21 +08:00
necks GlabelAveragePooling support 1d, 2d and 3d by param, and add neck test (#236) 2021-05-10 15:00:50 +08:00
utils [Fix] Use local ATTENTION registery to avoid conflict with other repo. (#375) 2021-07-27 17:59:05 +08:00
__init__.py Add classifiers, heads, necks and losses 2020-07-07 19:32:06 +08:00
builder.py [Fix] Use local ATTENTION registery to avoid conflict with other repo. (#375) 2021-07-27 17:59:05 +08:00