mirror of
https://github.com/alibaba/EasyCV.git
synced 2025-06-03 14:49:00 +08:00
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.
24 lines
858 B
Python
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
|