pytorch-image-models

Commit Graph

Author	SHA1	Message	Date
Ross Wightman	3b6cce4c95	Add initial impl of CrossStagePartial networks, yet to be trained, not quite the same as darknet cfgs.	4 years ago
Ross Wightman	eb7653614f	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	4 years ago

Author

SHA1

Message

Date

Ross Wightman

3b6cce4c95

Add initial impl of CrossStagePartial networks, yet to be trained, not quite the same as darknet cfgs.

Ross Wightman

eb7653614f

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

2 Commits (9214ca071674ce62b0eff36f0a1e3eaaba6ec2e3)