Skip to content

Commit 6f2eb1d

Browse files
authored
Update 2022-3-10-pytorch-1.11-new-library-releases.md
1 parent 3606d06 commit 6f2eb1d

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

_posts/2022-3-10-pytorch-1.11-new-library-releases.md

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -151,8 +151,8 @@ import torch
151151
from torchvision import models
152152

153153
x = torch.rand(1, 3, 224, 224)
154-
vit = models.detection.vit_b_16(pretrained=True).eval()
155-
convnext = models.detection.convnext_tiny(pretrained=True).eval()
154+
vit = models.vit_b_16(pretrained=True).eval()
155+
convnext = models.convnext_tiny(pretrained=True).eval()
156156
predictions1 = vit(x)
157157
predictions2 = convnext(x)
158158
```

0 commit comments

Comments
 (0)