You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
66 lines
1.9 KiB
66 lines
1.9 KiB
from .beit import *
|
|
from .byoanet import *
|
|
from .byobnet import *
|
|
from .cait import *
|
|
from .coat import *
|
|
from .convit import *
|
|
from .convmixer import *
|
|
from .convnext import *
|
|
from .crossvit import *
|
|
from .cspnet import *
|
|
from .deit import *
|
|
from .densenet import *
|
|
from .dla import *
|
|
from .dpn import *
|
|
from .efficientnet import *
|
|
from .ghostnet import *
|
|
from .gluon_resnet import *
|
|
from .gluon_xception import *
|
|
from .hardcorenas import *
|
|
from .hrnet import *
|
|
from .inception_resnet_v2 import *
|
|
from .inception_v3 import *
|
|
from .inception_v4 import *
|
|
from .levit import *
|
|
from .mlp_mixer import *
|
|
from .mobilenetv3 import *
|
|
from .mobilevit import *
|
|
from .nasnet import *
|
|
from .nest import *
|
|
from .nfnet import *
|
|
from .pit import *
|
|
from .pnasnet import *
|
|
from .poolformer import *
|
|
from .regnet import *
|
|
from .res2net import *
|
|
from .resnest import *
|
|
from .resnet import *
|
|
from .resnetv2 import *
|
|
from .rexnet import *
|
|
from .selecsls import *
|
|
from .senet import *
|
|
from .sequencer import *
|
|
from .sknet import *
|
|
from .swin_transformer import *
|
|
from .swin_transformer_v2_cr import *
|
|
from .tnt import *
|
|
from .tresnet import *
|
|
from .twins import *
|
|
from .vgg import *
|
|
from .visformer import *
|
|
from .vision_transformer import *
|
|
from .vision_transformer_hybrid import *
|
|
from .volo import *
|
|
from .vovnet import *
|
|
from .xception import *
|
|
from .xception_aligned import *
|
|
from .xcit import *
|
|
|
|
from .factory import create_model, parse_model_name, safe_model_name
|
|
from .helpers import load_checkpoint, resume_checkpoint, model_parameters
|
|
from .layers import TestTimePoolHead, apply_test_time_pool
|
|
from .layers import convert_splitbn_model
|
|
from .layers import is_scriptable, is_exportable, set_scriptable, set_exportable, is_no_jit, set_no_jit
|
|
from .registry import register_model, model_entrypoint, list_models, is_model, list_modules, is_model_in_modules,\
|
|
is_model_pretrained, get_pretrained_cfg, has_pretrained_cfg_key, is_pretrained_cfg_key, get_pretrained_cfg_value
|