vit-pytorch/vit_pytorch/vit.py at 8135d70e4e5ca292acaf7ee253324fa69a217d9c - vit-pytorch - Gitea: Git with a cup of tea

ZF/vit-pytorch

mirror of https://github.com/lucidrains/vit-pytorch.git synced 2025-12-30 08:02:29 +00:00

Files

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

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

3.9 KiB

Raw Blame History

View Raw