7 Commits

Author SHA1 Message Date
Ross Wightman
2a63d0246b Post merge cleanup 2021-06-07 14:38:30 -07:00
Mike
b87d98b238
Update convit.py
Cut out the duplicates
2021-06-06 17:58:31 +08: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
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