mirrors/mmclassification
1
0
Fork 0
mirror of https://github.com/open-mmlab/mmclassification.git synced 2025-06-03 21:53:55 +08:00
Code Issues Projects Releases Wiki Activity
mmclassification/mmpretrain/models/backbones
History
Ma Zerun b017670e1b
[Improve] Use PyTorch official scaled_dot_product_attention to accelerate MultiheadAttention. (#1434) 
* [Improve] Use PyTorch official `scaled_dot_product_attention` to accelerate `MultiheadAttention`.

* Support `--local-rank` and `--amp` option for new version PyTorch.

* Fix imports and UT.
2023-03-29 15:50:44 +08:00
..
__init__.py
[Refactor] Add self-supervised backbones and target generators. (#1379)
2023-02-28 15:59:17 +08:00
alexnet.py
Rename the package name to mmpretrain.
2023-02-17 15:20:55 +08:00
base_backbone.py
Rename the package name to mmpretrain.
2023-02-17 15:20:55 +08:00
beit.py
[Refactor] Use out_type to specify ViT-like backbone output. (#1408)
2023-03-09 11:02:58 +08:00
conformer.py
Rename the package name to mmpretrain.
2023-02-17 15:20:55 +08:00
convmixer.py
Rename the package name to mmpretrain.
2023-02-17 15:20:55 +08:00
convnext.py
Rename the package name to mmpretrain.
2023-02-17 15:20:55 +08:00
cspnet.py
Rename the package name to mmpretrain.
2023-02-17 15:20:55 +08:00
davit.py
Rename the package name to mmpretrain.
2023-02-17 15:20:55 +08:00
deit3.py
[Refactor] Use out_type to specify ViT-like backbone output. (#1408)
2023-03-09 11:02:58 +08:00
deit.py
[Refactor] Use out_type to specify ViT-like backbone output. (#1408)
2023-03-09 11:02:58 +08:00
densenet.py
Rename the package name to mmpretrain.
2023-02-17 15:20:55 +08:00
edgenext.py
Rename the package name to mmpretrain.
2023-02-17 15:20:55 +08:00
efficientformer.py
Rename the package name to mmpretrain.
2023-02-17 15:20:55 +08:00
efficientnet_v2.py
Rename the package name to mmpretrain.
2023-02-17 15:20:55 +08:00
efficientnet.py
Rename the package name to mmpretrain.
2023-02-17 15:20:55 +08:00
hornet.py
Rename the package name to mmpretrain.
2023-02-17 15:20:55 +08:00
hrnet.py
Rename the package name to mmpretrain.
2023-02-17 15:20:55 +08:00
inception_v3.py
Rename the package name to mmpretrain.
2023-02-17 15:20:55 +08:00
lenet.py
Rename the package name to mmpretrain.
2023-02-17 15:20:55 +08:00
levit.py
Rename the package name to mmpretrain.
2023-02-17 15:20:55 +08:00
mixmim.py
[Refactor] Use out_type to specify ViT-like backbone output. (#1408)
2023-03-09 11:02:58 +08:00
mlp_mixer.py
Rename the package name to mmpretrain.
2023-02-17 15:20:55 +08:00
mobilenet_v2.py
Rename the package name to mmpretrain.
2023-02-17 15:20:55 +08:00
mobilenet_v3.py
Rename the package name to mmpretrain.
2023-02-17 15:20:55 +08:00
mobileone.py
Rename the package name to mmpretrain.
2023-02-17 15:20:55 +08:00
mobilevit.py
Rename the package name to mmpretrain.
2023-02-17 15:20:55 +08:00
mvit.py
Rename the package name to mmpretrain.
2023-02-17 15:20:55 +08:00
poolformer.py
Rename the package name to mmpretrain.
2023-02-17 15:20:55 +08:00
regnet.py
Rename the package name to mmpretrain.
2023-02-17 15:20:55 +08:00
replknet.py
Rename the package name to mmpretrain.
2023-02-17 15:20:55 +08:00
repmlp.py
Rename the package name to mmpretrain.
2023-02-17 15:20:55 +08:00
repvgg.py
Rename the package name to mmpretrain.
2023-02-17 15:20:55 +08:00
res2net.py
Rename the package name to mmpretrain.
2023-02-17 15:20:55 +08:00
resnest.py
Rename the package name to mmpretrain.
2023-02-17 15:20:55 +08:00
resnet_cifar.py
Rename the package name to mmpretrain.
2023-02-17 15:20:55 +08:00
resnet.py
Rename the package name to mmpretrain.
2023-02-17 15:20:55 +08:00
resnext.py
Rename the package name to mmpretrain.
2023-02-17 15:20:55 +08:00
revvit.py
[Refactor] Use out_type to specify ViT-like backbone output. (#1408)
2023-03-09 11:02:58 +08:00
seresnet.py
Rename the package name to mmpretrain.
2023-02-17 15:20:55 +08:00
seresnext.py
Rename the package name to mmpretrain.
2023-02-17 15:20:55 +08:00
shufflenet_v1.py
Rename the package name to mmpretrain.
2023-02-17 15:20:55 +08:00
shufflenet_v2.py
Rename the package name to mmpretrain.
2023-02-17 15:20:55 +08:00
swin_transformer_v2.py
Rename the package name to mmpretrain.
2023-02-17 15:20:55 +08:00
swin_transformer.py
[Feature] Implement layer-wise learning rate decay optimizer constructor. (#1399)
2023-03-07 17:30:39 +08:00
t2t_vit.py
[Refactor] Use out_type to specify ViT-like backbone output. (#1408)
2023-03-09 11:02:58 +08:00
timm_backbone.py
Rename the package name to mmpretrain.
2023-02-17 15:20:55 +08:00
tinyvit.py
Rename the package name to mmpretrain.
2023-02-17 15:20:55 +08:00
tnt.py
Rename the package name to mmpretrain.
2023-02-17 15:20:55 +08:00
twins.py
[Improve] Use PyTorch official scaled_dot_product_attention to accelerate MultiheadAttention. (#1434)
2023-03-29 15:50:44 +08:00
van.py
Rename the package name to mmpretrain.
2023-02-17 15:20:55 +08:00
vgg.py
Rename the package name to mmpretrain.
2023-02-17 15:20:55 +08:00
vig.py
Rename the package name to mmpretrain.
2023-02-17 15:20:55 +08:00
vision_transformer.py
[Refactor] Use out_type to specify ViT-like backbone output. (#1408)
2023-03-09 11:02:58 +08:00
xcit.py
Rename the package name to mmpretrain.
2023-02-17 15:20:55 +08:00