Commit Graph

6 Commits (dbe7531aa3806154113c53bf22d189a72f70d11e)

Author SHA1 Message Date
Ross Wightman ab49d275de Significant norm update
3 years ago
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.
3 years ago
Ross Wightman 7ba5a384d3 Add ReXNet w/ remapped weights, feature support
4 years ago
Ross Wightman 1904ed8fec Improve dropblock impl, add fast variant, and better AMP speed, inplace, batchwise... few ResNeSt cleanups
4 years ago
Ross Wightman 208e7912f7 Missed one of the abalation model entrypoints, update README
4 years ago
Ross Wightman f4cdc2ac31 Add ResNeSt models
4 years ago