mmclassification/mmcls/models
Ma Zerun 33a4c0fe9f
[Fix] Use local ATTENTION registery to avoid conflict with other repo. (#375)
2021-07-27 17:59:05 +08:00
..
backbones Fix duplicate `init_weights` call in ViT init function. (#373) 2021-07-26 05:33:11 -04: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