Ross Wightman
|
2f5ed2dec1
|
Update `init_values` const for 24 and 36 layer ResMLP models
|
2021-06-07 17:15:04 -07:00 |
|
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 |