__init__.py 544 Bytes
Newer Older
1
from .alexnet import *
2
from .convnext import *
Geoff Pleiss's avatar
Geoff Pleiss committed
3
from .densenet import *
4
from .efficientnet import *
5
from .googlenet import *
6
from .inception import *
7
from .mnasnet import *
8
from .mobilenet import *
9
from .regnet import *
10
11
12
13
from .resnet import *
from .shufflenetv2 import *
from .squeezenet import *
from .vgg import *
14
from .vision_transformer import *
15
from .swin_transformer import *
16
from . import detection, optical_flow, quantization, segmentation, video
17
from ._api import get_model, get_model_builder, get_model_weights, get_weight, list_models