17 Commits (cb06c7a910cb9b1078679bc67c76afcbb7453d3c)

Author	SHA1	Message	Date
Ross Wightman	9811e229f7	Fix regression in models with 1001 class pretrained weights. Improve batchnorm arg and BatchNormAct layer handling in several models.	4 years ago
Ross Wightman	0167f749d3	Remove some old __future__ imports	4 years ago
Ross Wightman	b1f1a54de9	More uniform treatment of classifiers across all models, reduce code duplication.	4 years ago
Ross Wightman	6c17d57a2c	Fix some attributions, add copyrights to some file docstrings	4 years ago
Ross Wightman	3b9004bef9	Lots of changes to model creation helpers, close to finalizing feature extraction / interfaces	4 years ago
Ross Wightman	3aebc2f06c	Switch DPN to use BnAct layer, train a new DPN 68b model with RA to 79.21	4 years ago
Ross Wightman	88129b2569	Add set_layer_config contextmgr to adjust all layer configs at once, use in create_module with new args. Remove a few old warning causing constant annotations for jit.	5 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	5 years ago
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	5 years ago
Ross Wightman	13746a33fc	Big move, layer modules and fn to timm/models/layers	5 years ago
Ross Wightman	3bef524f9c	Finish with HRNet, weights and models updated. Improve consistency in model classifier/global pool treatment.	5 years ago
Ross Wightman	6057496409	Register dpn107	5 years ago
Ross Wightman	a6878b5218	Fix DPN config keys that I broke	5 years ago
Ross Wightman	b8762cc67d	Model updates. Add my best ResNet50 weights top-1=78.47. Add some other torchvision weights. ... * Remove some models that don't exist as pretrained an likely never will (se)resnext152 * Add some torchvision weights as tv_ for models that I have added better weights for * Add wide resnet recently added to torchvision along with resnext101-32x8d * Add functionality to model registry to allow filtering on pretrained weight presence	5 years ago
Ross Wightman	171c0b88b6	Add model registry and model listing fns, refactor model_factory/create_model fn	5 years ago
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	5 years ago
Ross Wightman	aa4354f466	Big re-org, working towards making pip/module as 'timm'	5 years ago