10 Commits

Author	SHA1	Message	Date
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	2020-06-01 17:16:52 -07:00
Vyacheslav Shults	a7ebe09029	Replace all None by nn.Identity() in all models reset_classifier when False-values num_classes is given. ... Make small code refactoring	2020-05-06 09:54:03 +03:00
Ross Wightman	ea300709f0	Fix model create fn not passing num_classes through. Fix #135	2020-05-01 18:37:03 -07:00
Ross Wightman	13746a33fc	Big move, layer modules and fn to timm/models/layers	2020-02-09 13:13:08 -08:00
Ross Wightman	3bef524f9c	Finish with HRNet, weights and models updated. Improve consistency in model classifier/global pool treatment.	2019-11-29 17:56:36 -08:00
Ross Wightman	171c0b88b6	Add model registry and model listing fns, refactor model_factory/create_model fn	2019-06-23 18:22:16 -07:00
Ross Wightman	87b92c528e	Some pretrianed URL changes ... * host some of Cadene's weights on github instead of .fr for speed * add my old port of ensemble adversarial inception resnet v2 * switch to my TF port of normal inception res v2 and change FC layer back to 'classif' for compat with ens_adv	2019-06-21 13:57:08 -07:00
Ross Wightman	63961b36a2	Missed pnasnet entrypoint	2019-06-20 23:34:20 -07:00
Ross Wightman	6cc214bd7a	Consistency in model entrypoints ... * move pretrained entrypoint arg to first pos to be closer to torchvision/hub * change DPN weight URLS to my github location	2019-06-20 23:29:44 -07:00
Ross Wightman	aa4354f466	Big re-org, working towards making pip/module as 'timm'	2019-06-19 17:20:51 -07:00