Ross Wightman
742c2d5247
Add Gather-Excite and Global Context attn modules. Refactor existing SE-like attn for consistency and refactor byob/byoanet for less redundancy.
2021-05-27 18:03:29 -07:00
Ross Wightman
c468c47a9c
Add regnety_160 weights from DeiT teacher model, update that and my regnety_032 weights to use higher test size.
2021-04-01 16:41:04 -07:00
Ross Wightman
288682796f
Update benchmark script to add precision arg. Fix some downstream (DeiT) compat issues with latest changes. Bump version to 0.4.7
2021-04-01 16:40:12 -07:00
Ross Wightman
d584e7f617
Support for huggingface hub via create_model and default_cfgs.
...
* improve consistency of model creation helper fns
* add comments to some of the model helpers
* support passing external default_cfgs so they can be sourced from hub
2021-03-16 22:48:26 -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
512b2dd645
Add new EfficientNet-B3 and RegNetY-3.2GF weights, both just over 82 top-1
2020-08-11 14:18:51 -07:00
Ross Wightman
6890300877
Add DropPath (stochastic depth) to RegNet
2020-08-11 14:08:53 -07:00
Ross Wightman
6c17d57a2c
Fix some attributions, add copyrights to some file docstrings
2020-07-27 13:44:56 -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
a66df5fb91
More model feature extraction support, start to deprecate senet.py, dilations added to regnet, add proper aligned xception
2020-07-03 00:41:30 -07:00
Ross Wightman
0aca08384f
Update regnet cfg keys to match model names so registry works properly
2020-06-09 14:37:45 -07:00
Ross Wightman
50658b9a67
Add RegNet models and weights
2020-05-18 00:08:52 -07:00