7 Commits (4547920f859b9d9ddfe630800ae375c4bcc7b3b6)

Author	SHA1	Message	Date
Ross Wightman	ab49d275de	Significant norm update ... * ConvBnAct layer renamed -> ConvNormAct and ConvNormActAa for anti-aliased * Significant update to EfficientNet and MobileNetV3 arch to support NormAct layers and grouped conv (as alternative to depthwise) * Update RegNet to add Z variant * Add Pre variant of XceptionAligned that works with NormAct layers * EvoNorm matches bits_and_tpu branch for merge	2021-12-14 13:48:30 -08:00
Alexander Soare	b25ff96768	wip - pre-rebase	2021-11-12 20:45:05 +00:00
Alexander Soare	e051dce354	Make all models FX traceable	2021-11-12 20:45:05 +00:00
Alexander Soare	0149ec30d7	wip - attempting to rebase	2021-11-12 20:45:05 +00:00
Alexander Soare	bc3d4eb403	wip -rebase	2021-11-12 20:45:05 +00:00
Ross Wightman	b7a568f065	Fix torchscript issue in bat	2021-06-08 23:19:51 -07:00
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