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
7ef7788ee9
Fix CUDA crash w/ channels-last + CSP models. Remove use of chunk()
2021-03-06 16:56:34 -08:00
Ross Wightman
0f5d9d8166
Add CSPResNet50 weights, 79.6 top-1 at 256x256
2020-08-12 11:20:11 -07:00
Ross Wightman
d72ddafe56
Fix some checkpoint / model str regressions
2020-07-29 19:43:01 -07:00
Ross Wightman
a69c0e04f0
Fix pool size in cspnet
2020-07-27 13:44:02 -07:00
Ross Wightman
14ef7a0dd6
Rename csp.py -> cspnet.py
2020-07-27 11:15:07 -07:00