pytorch-image-models/timm/models/__init__.py

31 lines
877 B
Python
Raw Normal View History

from .inception_v4 import *
from .inception_resnet_v2 import *
from .densenet import *
from .resnet import *
from .dpn import *
from .senet import *
from .xception import *
2019-07-04 13:10:50 +08:00
from .nasnet import *
from .pnasnet import *
2019-12-29 03:41:55 +08:00
from .selecsls import *
from .efficientnet import *
from .mobilenetv3 import *
from .inception_v3 import *
from .gluon_resnet import *
from .gluon_xception import *
from .res2net import *
from .dla import *
from .hrnet import *
from .sknet import *
2020-04-12 23:44:12 +08:00
from .tresnet import *
2020-05-12 14:27:09 +08:00
from .resnest import *
2020-05-18 15:08:52 +08:00
from .regnet import *
from .vovnet import *
from .registry import *
from .factory import create_model
from .helpers import load_checkpoint, resume_checkpoint
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