164 Commits

Author	SHA1	Message	Date
Ross Wightman	b1f1a54de9	More uniform treatment of classifiers across all models, reduce code duplication.	2020-08-03 22:18:24 -07:00
Ross Wightman	d72ddafe56	Fix some checkpoint / model str regressions	2020-07-29 19:43:01 -07:00
Ross Wightman	ea58e0bc16	Disable big models for MacOS test since they are starting to fail frequently.	2020-07-27 18:01:43 -07:00
Ross Wightman	7ba5a384d3	Add ReXNet w/ remapped weights, feature support	2020-07-23 10:28:57 -07:00
Ross Wightman	c9d54bc1c3	Add HRNet feature extraction, fix senet type, lower feature testing res to 96x96	2020-07-21 17:39:29 -07:00
Ross Wightman	2ac663f340	Add feature support to legacy senets, add 32x32 resnext models to exclude list for feature testing.	2020-07-21 11:15:30 -07:00
Ross Wightman	4e61c6a12d	Cleanup, refactoring of Feature extraction code, add tests, fix tests, non hook feature extraction working with torchscript	2020-07-20 16:10:31 -07:00
Ross Wightman	9eba134d79	More models supporting feature extraction, xception, gluon_xception, inception_v3, inception_v4, pnasnet, nasnet, dla. Fix DLA unused projection params.	2020-07-19 14:02:02 -07:00
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
Ross Wightman	4d13db538f	Add x48d ResNext101s to test exclude for ubuntu	2020-05-21 19:13:41 -07:00
Ross Wightman	4212cd3b9f	Another attempt at getting Ubuntu test runner to work	2020-05-21 18:55:10 -07:00
Ross Wightman	20329f2630	Bring down test resolutions to see if we can at least do a fwd on the L2 models	2020-05-21 16:49:46 -07:00
Ross Wightman	3873ea710e	Minor test change	2020-05-21 15:51:47 -07:00
Ross Wightman	afb6bd0669	Add backward and default_cfg tests and fix a few issues found. Fix #153	2020-05-21 15:30:58 -07:00