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
4 lines
189 B
Python
4 lines
189 B
Python
from .version import __version__
|
|
from .models import create_model, list_models, is_model, list_modules, model_entrypoint, \
|
|
is_scriptable, is_exportable, set_scriptable, set_exportable
|