from .alexnet import AlexNet
from .caformer import CAFormer_B36, CAFormer_M36, CAFormer_S18, CAFormer_S36
from .coatnet import (
    CoAtNet_0_RW_224,
    CoAtNet_1_RW_224,
    CoAtNet_2_RW_224,
)
from .convformer import ConvFormer_B36, ConvFormer_M36, ConvFormer_S18, ConvFormer_S36
from .convnext import ConvNeXt_Base, ConvNeXt_Large, ConvNeXt_Small, ConvNeXt_Tiny
from .deit import (
    DeiT_Base_Patch16_224,
    DeiT_Base_Patch16_384,
    DeiT_Small_Patch16_224,
    DeiT_Tiny_Patch16_224,
)
from .deit3 import (
    DeiT3_Base_Patch16_224,
    DeiT3_Base_Patch16_384,
    DeiT3_Large_Patch16_224,
    DeiT3_Large_Patch16_384,
    DeiT3_Medium_Patch16_224,
    DeiT3_Small_Patch16_224,
    DeiT3_Small_Patch16_384,
)
from .densenet import DenseNet121, DenseNet161, DenseNet169, DenseNet201
from .efficientformer import EfficientFormer_L1, EfficientFormer_L3, EfficientFormer_L7
from .efficientnet import (
    EfficientNet_B0,
    EfficientNet_B1,
    EfficientNet_B2,
    EfficientNet_B3,
    EfficientNet_B4,
    EfficientNet_B5,
    EfficientNet_B6,
    EfficientNet_B7,
)
from .efficientnet_v2 import (
    EfficientNet_V2_L,
    EfficientNet_V2_M,
    EfficientNet_V2_S,
)
from .flexivit import FlexiViT_Base, FlexiViT_Large, FlexiViT_Small
from .googlenet import GoogLeNet
from .inception_v3 import Inception_V3
from .levit import (
    LeViT_Conv_128,
    LeViT_Conv_128S,
    LeViT_Conv_192,
    LeViT_Conv_256,
    LeViT_Conv_384,
)
from .mnasnet import MNASNet0_5, MNASNet0_75, MNASNet1_0, MNASNet1_3
from .mobilenet_v2 import MobileNet_V2
from .mobilenet_v3 import MobileNet_V3_Large, MobileNet_V3_Small
from .regnet import (
    RegNet_X_1_6GF,
    RegNet_X_3_2GF,
    RegNet_X_8GF,
    RegNet_X_16GF,
    RegNet_X_32GF,
    RegNet_X_400MF,
    RegNet_X_800MF,
    RegNet_Y_1_6GF,
    RegNet_Y_3_2GF,
    RegNet_Y_8GF,
    RegNet_Y_16GF,
    RegNet_Y_32GF,
    RegNet_Y_400MF,
    RegNet_Y_800MF,
)
from .repvit import (
    RepViT_M0_9,
    RepViT_M1,
    RepViT_M1_0,
    RepViT_M1_1,
    RepViT_M1_5,
    RepViT_M2,
    RepViT_M2_3,
    RepViT_M3,
)
from .resnet import ResNet18, ResNet34, ResNet50, ResNet101, ResNet152
from .resnext import ResNeXt50_32x4d, ResNeXt101_32x8d, ResNeXt101_64x4d
from .shufflenet_v2 import ShuffleNet_V2_X1_0, ShuffleNet_V2_X1_5, ShuffleNet_V2_X2_0
from .squeezenet import SqueezeNet1_0, SqueezeNet1_1
from .swin import Swin_B, Swin_S, Swin_T
from .vgg import VGG11, VGG11_BN, VGG13, VGG13_BN, VGG16, VGG16_BN, VGG19, VGG19_BN
from .visformer import VisFormer_Small, VisFormer_Tiny
from .vit import (
    ViT_B_16,
    ViT_B_32,
    ViT_Base_Patch8_224,
    ViT_Base_Patch16_224,
    ViT_Base_Patch16_384,
    ViT_Base_Patch32_224,
    ViT_Base_Patch32_384,
    ViT_L_16,
    ViT_L_32,
    ViT_Large_Patch16_224,
    ViT_Large_Patch16_384,
    ViT_Large_Patch32_384,
    ViT_Small_Patch16_224,
    ViT_Small_Patch16_384,
    ViT_Small_Patch32_224,
    ViT_Small_Patch32_384,
    ViT_Tiny_Patch16_224,
    ViT_Tiny_Patch16_384,
)
from .wide_resnet import Wide_ResNet50_2, Wide_ResNet101_2
from .yolo11_cls import YOLO11lCls, YOLO11mCls, YOLO11nCls, YOLO11sCls, YOLO11xCls
from .yolo26_cls import YOLO26lCls, YOLO26mCls, YOLO26nCls, YOLO26sCls, YOLO26xCls
from .yolov5_cls import YOLOv5lCls, YOLOv5mCls, YOLOv5nCls, YOLOv5sCls, YOLOv5xCls
from .yolov8_cls import YOLOv8lCls, YOLOv8mCls, YOLOv8nCls, YOLOv8sCls, YOLOv8xCls

__all__ = [
    "AlexNet",
    "CAFormer_B36",
    "CAFormer_M36",
    "CAFormer_S18",
    "CAFormer_S36",
    "CoAtNet_0_RW_224",
    "CoAtNet_1_RW_224",
    "CoAtNet_2_RW_224",
    "ConvFormer_B36",
    "ConvFormer_M36",
    "ConvFormer_S18",
    "ConvFormer_S36",
    "ConvNeXt_Base",
    "ConvNeXt_Large",
    "ConvNeXt_Small",
    "ConvNeXt_Tiny",
    "DeiT_Base_Patch16_224",
    "DeiT_Base_Patch16_384",
    "DeiT_Small_Patch16_224",
    "DeiT_Tiny_Patch16_224",
    "DeiT3_Base_Patch16_224",
    "DeiT3_Base_Patch16_384",
    "DeiT3_Large_Patch16_224",
    "DeiT3_Large_Patch16_384",
    "DeiT3_Medium_Patch16_224",
    "DeiT3_Small_Patch16_224",
    "DeiT3_Small_Patch16_384",
    "DenseNet121",
    "DenseNet161",
    "DenseNet169",
    "DenseNet201",
    "EfficientFormer_L1",
    "EfficientFormer_L3",
    "EfficientFormer_L7",
    "EfficientNet_B0",
    "EfficientNet_B1",
    "EfficientNet_B2",
    "EfficientNet_B3",
    "EfficientNet_B4",
    "EfficientNet_B5",
    "EfficientNet_B6",
    "EfficientNet_B7",
    "EfficientNet_V2_L",
    "EfficientNet_V2_M",
    "EfficientNet_V2_S",
    "FlexiViT_Base",
    "FlexiViT_Large",
    "FlexiViT_Small",
    "GoogLeNet",
    "Inception_V3",
    "LeViT_Conv_128",
    "LeViT_Conv_128S",
    "LeViT_Conv_192",
    "LeViT_Conv_256",
    "LeViT_Conv_384",
    "MNASNet0_5",
    "MNASNet0_75",
    "MNASNet1_0",
    "MNASNet1_3",
    "MobileNet_V2",
    "MobileNet_V3_Large",
    "MobileNet_V3_Small",
    "RegNet_X_1_6GF",
    "RegNet_X_3_2GF",
    "RegNet_X_8GF",
    "RegNet_X_16GF",
    "RegNet_X_32GF",
    "RegNet_X_400MF",
    "RegNet_X_800MF",
    "RegNet_Y_1_6GF",
    "RegNet_Y_3_2GF",
    "RegNet_Y_8GF",
    "RegNet_Y_16GF",
    "RegNet_Y_32GF",
    "RegNet_Y_400MF",
    "RegNet_Y_800MF",
    "RepViT_M0_9",
    "RepViT_M1",
    "RepViT_M1_0",
    "RepViT_M1_1",
    "RepViT_M1_5",
    "RepViT_M2",
    "RepViT_M2_3",
    "RepViT_M3",
    "ResNet18",
    "ResNet34",
    "ResNet50",
    "ResNet101",
    "ResNet152",
    "ResNeXt50_32x4d",
    "ResNeXt101_32x8d",
    "ResNeXt101_64x4d",
    "ShuffleNet_V2_X1_0",
    "ShuffleNet_V2_X1_5",
    "ShuffleNet_V2_X2_0",
    "SqueezeNet1_0",
    "SqueezeNet1_1",
    "Swin_B",
    "Swin_S",
    "Swin_T",
    "VGG11",
    "VGG11_BN",
    "VGG13",
    "VGG13_BN",
    "VGG16",
    "VGG16_BN",
    "VGG19",
    "VGG19_BN",
    "VisFormer_Small",
    "VisFormer_Tiny",
    "ViT_B_16",
    "ViT_B_32",
    "ViT_Base_Patch8_224",
    "ViT_Base_Patch16_224",
    "ViT_Base_Patch16_384",
    "ViT_Base_Patch32_224",
    "ViT_Base_Patch32_384",
    "ViT_L_16",
    "ViT_L_32",
    "ViT_Large_Patch16_224",
    "ViT_Large_Patch16_384",
    "ViT_Large_Patch32_384",
    "ViT_Small_Patch16_224",
    "ViT_Small_Patch16_384",
    "ViT_Small_Patch32_224",
    "ViT_Small_Patch32_384",
    "ViT_Tiny_Patch16_224",
    "ViT_Tiny_Patch16_384",
    "Wide_ResNet50_2",
    "Wide_ResNet101_2",
    "YOLO11lCls",
    "YOLO11mCls",
    "YOLO11nCls",
    "YOLO11sCls",
    "YOLO11xCls",
    "YOLO26lCls",
    "YOLO26mCls",
    "YOLO26nCls",
    "YOLO26sCls",
    "YOLO26xCls",
    "YOLOv5lCls",
    "YOLOv5mCls",
    "YOLOv5nCls",
    "YOLOv5sCls",
    "YOLOv5xCls",
    "YOLOv8lCls",
    "YOLOv8mCls",
    "YOLOv8nCls",
    "YOLOv8sCls",
    "YOLOv8xCls",
]
