9 Commits

Author	SHA1	Message	Date
Ross Wightman	5a8e1e643e	Initial Normalizer-Free Reg/ResNet impl. A bit of related layer refactoring.	2021-01-27 22:06:57 -08:00
Ross Wightman	f31933cb37	Initial Vision Transformer impl w/ patch and hybrid variants. Refactor tuple helpers.	2020-10-13 13:33:44 -07:00
Ross Wightman	e8e2d9cabf	Add DropPath (stochastic depth) to ReXNet and VoVNet. RegNet DropPath impl tweak and dedupe se args.	2020-09-24 18:20:36 -07:00
Ross Wightman	e8ca45854c	More models in sotabench, more control over sotabench run, dataset filename extraction consistency	2020-09-24 15:56:57 -07:00
Ross Wightman	110a7c4982	AdaptiveAvgPool2d -> mean((2,3)) for all SE/attn layers to avoid NaN with AMP + channels_last layout. See https://github.com/pytorch/pytorch/issues/43992	2020-09-01 16:05:32 -07:00
Ross Wightman	b1f1a54de9	More uniform treatment of classifiers across all models, reduce code duplication.	2020-08-03 22:18:24 -07:00
Ross Wightman	ac18adb9c3	Remove debug print from RexNet	2020-07-29 11:15:19 -07:00
Ross Wightman	6c17d57a2c	Fix some attributions, add copyrights to some file docstrings	2020-07-27 13:44:56 -07:00
Ross Wightman	7ba5a384d3	Add ReXNet w/ remapped weights, feature support	2020-07-23 10:28:57 -07:00