zzoneee 0cb91de0cb
add DeiT III (#171)
1.Add a backbone: deitiii.
2.Add an optimizer: lamb. 
3.Add a sampler: RASampler. 
4.Add a lr update hook: CosineAnnealingWarmupByEpochLrUpdaterHook.
5.In easycv/models/classification/classification.py, I remove the default mixup_cfg to keep the classification.py clean.
2022-09-14 15:24:54 +08:00

24 lines
858 B
Python

# Copyright (c) Alibaba, Inc. and its affiliates.
from .benchmark_mlp import BenchMarkMLP
from .bninception import BNInception
from .conv_mae_vit import FastConvMAEViT
from .conv_vitdet import ConvViTDet
from .efficientformer import EfficientFormer
from .genet import PlainNet
from .hrnet import HRNet
from .inceptionv3 import Inception3
from .lighthrnet import LiteHRNet
from .mae_vit_transformer import *
from .mnasnet import MNASNet
from .mobilenetv2 import MobileNetV2
from .pytorch_image_models_wrapper import *
from .repvgg_yolox_backbone import RepVGGYOLOX
from .resnest import ResNeSt
from .resnet import ResNet
from .resnet_jit import ResNetJIT
from .resnext import ResNeXt
from .shuffle_transformer import ShuffleTransformer
from .swin_transformer import SwinTransformer
from .vision_transformer import VisionTransformer
from .vitdet import ViTDet