mirror of
https://github.com/huggingface/pytorch-image-models.git
synced 2025-06-03 15:01:08 +08:00
eb7653614f
* refactor activations into basic PyTorch, jit scripted, and memory efficient custom auto * implement hard-mish, better grad for hard-swish * add initial VovNet V1/V2 impl, fix #151 * VovNet and DenseNet first models to use NormAct layers (support BatchNormAct2d, EvoNorm, InplaceIABN) * Wrap IABN for any models that use it * make more models torchscript compatible (DPN, PNasNet, Res2Net, SelecSLS) and add tests
31 lines
877 B
Python
31 lines
877 B
Python
from .inception_v4 import *
|
|
from .inception_resnet_v2 import *
|
|
from .densenet import *
|
|
from .resnet import *
|
|
from .dpn import *
|
|
from .senet import *
|
|
from .xception import *
|
|
from .nasnet import *
|
|
from .pnasnet import *
|
|
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 *
|
|
from .tresnet import *
|
|
from .resnest import *
|
|
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
|