Skip to content

Commit

Permalink
add link to official repo for levit
Browse files Browse the repository at this point in the history
  • Loading branch information
lucidrains committed May 6, 2021
1 parent 74b4023 commit 7333979
Showing 1 changed file with 2 additions and 0 deletions.
2 changes: 2 additions & 0 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -271,6 +271,8 @@ preds = v(img) # (1, 1000)

<a href="https://arxiv.org/abs/2104.01136">This paper</a> proposes a number of changes, including (1) convolutional embedding instead of patch-wise projection (2) downsampling in stages (3) extra non-linearity in attention (4) 2d relative positional biases instead of initial absolute positional bias (5) batchnorm in place of layernorm.

<a href="https://github.com/facebookresearch/LeViT">Official repository</a>

```python
import torch
from vit_pytorch.levit import LeViT
Expand Down

0 comments on commit 7333979

Please sign in to comment.