14 Commits (74fcd4c50e54d4e9a45c835564bfaccb713c40ad)

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	231d04e91a	ResNetV2 pre-act and non-preact model, w/ BiT pretrained weights and support for ViT R50 model. Tweaks for in21k num_classes passing. More to do... tests failing.	4 years ago
Ross Wightman	b1f1a54de9	More uniform treatment of classifiers across all models, reduce code duplication.	4 years ago
Ross Wightman	4e61c6a12d	Cleanup, refactoring of Feature extraction code, add tests, fix tests, non hook feature extraction working with torchscript	4 years ago
Ross Wightman	9eba134d79	More models supporting feature extraction, xception, gluon_xception, inception_v3, inception_v4, pnasnet, nasnet, dla. Fix DLA unused projection params.	4 years ago
Ross Wightman	a66df5fb91	More model feature extraction support, start to deprecate senet.py, dilations added to regnet, add proper aligned xception	4 years ago
Ross Wightman	d0113f9cdb	Fix a few issues that came up in tests	4 years ago
Ross Wightman	d23a2697d0	Working on feature extraction, interfaces refined, a number of models working, some in progress.	4 years ago
Ross Wightman	0ea53cecc3	Merge branch 'master' into densenet_update_and_more	5 years ago
Ross Wightman	afb6bd0669	Add backward and default_cfg tests and fix a few issues found. Fix #153	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	da52fcf78a	Add NASNet-Large model	5 years ago