Phil Wang 2cb6b35030 complete levit

2021-04-06 13:36:11 -07:00

..

__init__.py

add deep vit

2021-03-23 11:57:13 -07:00

cait.py

add layer dropout for CaiT

2021-04-01 20:30:37 -07:00

cross_vit.py

add documentation for three recent vision transformer follow-up papers

2021-03-31 09:22:15 -07:00

cvt.py

add documentation for three recent vision transformer follow-up papers

2021-03-31 09:22:15 -07:00

deepvit.py

add deep vit

2021-03-23 11:57:13 -07:00

distill.py

offer hard distillation as well

2021-04-01 16:56:14 -07:00

efficient.py

make sure distillation still works

2021-02-21 19:08:18 -08:00

levit.py

complete levit

2021-04-06 13:36:11 -07:00

mpp.py

cleanup and release 0.8.0

2021-03-08 07:28:31 -08:00

pit.py

add documentation for three recent vision transformer follow-up papers

2021-03-31 09:22:15 -07:00

recorder.py

make recorder work for t2t and deepvit

2021-03-29 18:16:34 -07:00

t2t.py

remove masking, as it complicates with little benefit

2021-03-23 12:18:47 -07:00

vit.py

use hooks to retrieve attention maps for user without modifying ViT

2021-03-29 15:10:12 -07:00