Ross Wightman
|
307a935b79
|
Add non-local and BAT attention. Merge attn and self-attn factories into one. Add attention references to README. Add mlp 'mode' to ECA.
|
2021-05-31 13:18:11 -07:00 |
|
Ross Wightman
|
7ba5a384d3
|
Add ReXNet w/ remapped weights, feature support
|
2020-07-23 10:28:57 -07:00 |
|
Ross Wightman
|
1904ed8fec
|
Improve dropblock impl, add fast variant, and better AMP speed, inplace, batchwise... few ResNeSt cleanups
|
2020-05-13 15:17:08 -07:00 |
|
Ross Wightman
|
208e7912f7
|
Missed one of the abalation model entrypoints, update README
|
2020-05-12 13:36:31 -07:00 |
|
Ross Wightman
|
f4cdc2ac31
|
Add ResNeSt models
|
2020-05-11 23:27:09 -07:00 |