11 Commits

Author	SHA1	Message	Date
Yusuke Uchida	f6b56602f9	fix test_model_default_cfgs	2020-08-11 23:23:57 +09: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	3b9004bef9	Lots of changes to model creation helpers, close to finalizing feature extraction / interfaces	2020-07-17 17:54:26 -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	0ea53cecc3	Merge branch 'master' into densenet_update_and_more	2020-05-22 16:18:10 -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
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	be7c784d21	Small TResNet simplification, just use SelectAdaptivePool, don't notice any perf difference	2020-04-27 17:50:19 -07:00
Ross Wightman	0004f37d25	Some TResNet cleanup. ... * allow use of global pool arg, test-time-pooling * clean checkpoints to just contain state dict, add 448 res checkpoints * support DataParallel via lazy filter creation for JIT Downsample * some minor formatting (mostly alignment) preferences	2020-04-27 13:21:05 -07:00
talrid	8a63c1add8	finalizing	2020-04-12 19:08:35 +03:00
talrid	6209146738	TResNet models	2020-04-12 18:44:12 +03:00