"torchvision/csrc/cpu/video_reader/Interface.h" did not exist on "a6a926bc8bdb8a32d910943c9c78e231879ff989"
__init__.py 566 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 *
Ponku's avatar
Ponku committed
16
from .maxvit import *
17
from . import detection, optical_flow, quantization, segmentation, video
18
from ._api import get_model, get_model_builder, get_model_weights, get_weight, list_models