Phil Wang 8135d70e4e use hooks to retrieve attention maps for user without modifying ViT

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

..

__init__.py

add deep vit

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

deepvit.py

add deep vit

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

distill.py

remove masking, as it complicates with little benefit

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

efficient.py

make sure distillation still works

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

mpp.py

cleanup and release 0.8.0

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

recorder.py

use hooks to retrieve attention maps for user without modifying ViT

2021-03-29 15:10:12 -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