Skip to content

Commit

Permalink
Update Readme.md
Browse files Browse the repository at this point in the history
  • Loading branch information
jacobgil authored Dec 31, 2020
1 parent 1d68f22 commit 15a81d3
Showing 1 changed file with 4 additions and 1 deletion.
5 changes: 4 additions & 1 deletion Readme.md
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,9 @@

This repository implements methods for explainability in Vision Transformers.

See also https://jacobgil.github.io/deeplearning/vision-transformer-explainability


## Currently implemented:
- Attention Rollout.
- Gradient Attention Rollout for class specific explainability.
Expand Down Expand Up @@ -96,4 +99,4 @@ and especially [the vision transformer implementation](https://github.com/rwight
- Credit for https://github.com/jeonsworld/ViT-pytorch for being a good starting point.

## Requirements
`pip install timm`
`pip install timm`

0 comments on commit 15a81d3

Please sign in to comment.