7 Commits

Author	SHA1	Message	Date
Ross Wightman	bfc72f75d3	Expand scope of testing for non-std vision transformer / mlp models. Some related cleanup and create fn cleanup for all vision transformer and mlp models. More CoaT weights.	2021-05-24 21:13:26 -07:00
talrid	dc1a4efd28	mixer_b16_224_miil, mixer_b16_224_miil_in21k models	2021-05-20 10:35:50 +03:00
Ross Wightman	d5af752117	Add preliminary gMLP and ResMLP impl to Mlp-Mixer	2021-05-19 09:55:05 -07:00
Ross Wightman	e7f0db8664	Fix drop/drop_path arg on MLP-Mixer model. Fix #641	2021-05-16 08:31:52 -07:00
Ross Wightman	b2c305c2aa	Move Mlp and PatchEmbed modules into layers. Being used in lots of models now...	2021-05-06 14:03:23 -07:00
Ross Wightman	2d8b09fe8b	Add official pretrained weights to MLP-Mixer, complete model cfgs.	2021-05-05 15:59:40 -07:00
Ross Wightman	12efffa6b1	Initial MLP-Mixer attempt...	2021-05-05 00:59:45 -07:00