ZF/vit-pytorch
1
0
Fork 0
mirror of https://github.com/lucidrains/vit-pytorch.git synced 2025-12-30 08:02:29 +00:00
Code Issues Packages Projects Releases Wiki Activity

readme

Browse Source
This commit is contained in:
Phil Wang
2021-11-07 09:59:49 -08:00
parent 06d375351e
commit 9e50b2a41e
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
README.md
View File

@@ -458,8 +458,9 @@ model = RegionViT(
use_peg = False, # whether to use positional generating module. they used this for object detection for a boost in performance
)
x = torch.randn(1, 3, 224, 224)
logits = model(x) # (1, 1000)
img = torch.randn(1, 3, 224, 224)
pred = model(img) # (1, 1000)
```
## NesT
@@ -485,6 +486,7 @@ nest = NesT(
)
img = torch.randn(1, 3, 224, 224)
pred = nest(img) # (1, 1000)
```