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 |
|
Ross Wightman
|
b7de82e835
|
ConViT cleanup, fix torchscript, bit of reformatting, reuse existing layers.
|
2021-05-21 17:04:23 -07:00 |
Aman Arora
|
5db1eb6ba5
|
Add defaults
|
2021-05-21 02:11:20 +00:00 |
|
Aman Arora
|
8b1f2e8e1f
|
remote unused matplotlib import
|
2021-05-20 23:42:42 +00:00 |
|
Aman Arora
|
40c506ba1e
|
Add ConViT
|
2021-05-20 23:17:28 +00:00 |