pytorch-image-models

mirror of https://github.com/huggingface/pytorch-image-models.git

Author	SHA1	Message	Date
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