Ross Wightman
|
317ea3e599
|
Tweak NFNet, ResNetV2, RexNet feature extraction, use pre-act features for simplicty. Update tests for pytorch 1.8. 0.4.5 version bump.
|
2021-03-06 12:48:06 -08:00 |
|
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 |