2021-02-09 16:21:47 -08:00
|
|
|
from .byobnet import *
|
2020-07-27 11:15:07 -07:00
|
|
|
from .cspnet import *
|
2019-06-23 18:22:16 -07:00
|
|
|
from .densenet import *
|
2020-07-03 00:41:30 -07:00
|
|
|
from .dla import *
|
2019-06-23 18:22:16 -07:00
|
|
|
from .dpn import *
|
2019-11-28 16:55:36 -08:00
|
|
|
from .efficientnet import *
|
2019-06-23 18:22:16 -07:00
|
|
|
from .gluon_resnet import *
|
2019-08-10 13:52:01 -07:00
|
|
|
from .gluon_xception import *
|
2019-11-25 22:34:04 -08:00
|
|
|
from .hrnet import *
|
2020-07-03 00:41:30 -07:00
|
|
|
from .inception_resnet_v2 import *
|
|
|
|
from .inception_v3 import *
|
|
|
|
from .inception_v4 import *
|
|
|
|
from .mobilenetv3 import *
|
|
|
|
from .nasnet import *
|
2021-01-27 22:06:57 -08:00
|
|
|
from .nfnet import *
|
2020-07-03 00:41:30 -07:00
|
|
|
from .pnasnet import *
|
|
|
|
from .regnet import *
|
|
|
|
from .res2net import *
|
|
|
|
from .resnest import *
|
|
|
|
from .resnet import *
|
2020-12-24 12:03:31 -08:00
|
|
|
from .resnetv2 import *
|
2020-07-23 10:28:57 -07:00
|
|
|
from .rexnet import *
|
2020-07-03 00:41:30 -07:00
|
|
|
from .selecsls import *
|
|
|
|
from .senet import *
|
2020-01-26 11:23:39 -08:00
|
|
|
from .sknet import *
|
2020-04-12 18:44:12 +03:00
|
|
|
from .tresnet import *
|
2021-02-09 16:21:47 -08:00
|
|
|
from .vgg import *
|
2020-10-13 13:33:44 -07:00
|
|
|
from .vision_transformer import *
|
Monster commit, activation refactor, VoVNet, norm_act improvements, more
* 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
2020-06-01 16:59:51 -07:00
|
|
|
from .vovnet import *
|
2020-07-03 00:41:30 -07:00
|
|
|
from .xception import *
|
|
|
|
from .xception_aligned import *
|
2021-03-14 22:06:33 +02:00
|
|
|
from .hardcorenas import *
|
2021-03-28 19:53:42 +08:00
|
|
|
from .tnt import *
|
2019-06-23 18:22:16 -07:00
|
|
|
|
2021-03-16 22:48:26 -07:00
|
|
|
from .factory import create_model, split_model_name, safe_model_name
|
2021-02-15 23:22:44 -08:00
|
|
|
from .helpers import load_checkpoint, resume_checkpoint, model_parameters
|
2020-02-09 13:13:08 -08:00
|
|
|
from .layers import TestTimePoolHead, apply_test_time_pool
|
|
|
|
from .layers import convert_splitbn_model
|
Monster commit, activation refactor, VoVNet, norm_act improvements, more
* 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
2020-06-01 16:59:51 -07:00
|
|
|
from .layers import is_scriptable, is_exportable, set_scriptable, set_exportable, is_no_jit, set_no_jit
|
2020-07-03 00:41:30 -07:00
|
|
|
from .registry import *
|
2021-03-14 22:06:33 +02:00
|
|
|
|