5 Commits (9c406532bde4ffe281d356de6e597717d2e53205)

Author	SHA1	Message	Date
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	4 years ago
Ross Wightman	b1f1a54de9	More uniform treatment of classifiers across all models, reduce code duplication.	4 years ago
Ross Wightman	ac18adb9c3	Remove debug print from RexNet	4 years ago
Ross Wightman	6c17d57a2c	Fix some attributions, add copyrights to some file docstrings	4 years ago
Ross Wightman	7ba5a384d3	Add ReXNet w/ remapped weights, feature support	4 years ago