2019-06-19 17:19:37 -07:00
|
|
|
from .version import __version__
|
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 .models import create_model, list_models, is_model, list_modules, model_entrypoint, \
|
2021-04-12 11:54:22 -07:00
|
|
|
is_scriptable, is_exportable, set_scriptable, set_exportable, has_model_default_key, is_model_default_key, \
|
|
|
|
|
get_model_default_value, is_model_pretrained
|
