Skip to content
Snippets Groups Projects
Unverified Commit a15f6afe authored by Couffou's avatar Couffou Committed by GitHub
Browse files

Merge pull request #6 from xhlulu/main

Fix torch hub link
parents 6f51bb16 1041dd07
Branches
No related tags found
No related merge requests found
...@@ -78,11 +78,11 @@ You can choose to download only the weights of the pretrained backbone used for ...@@ -78,11 +78,11 @@ You can choose to download only the weights of the pretrained backbone used for
The pretrained models are available on PyTorch Hub. The pretrained models are available on PyTorch Hub.
```python ```python
import torch import torch
deits16 = torch.hub.load('facebookresearch/dino', 'dino_deits16') deits16 = torch.hub.load('facebookresearch/dino:main', 'dino_deits16')
deits8 = torch.hub.load('facebookresearch/dino', 'dino_deits8') deits8 = torch.hub.load('facebookresearch/dino:main', 'dino_deits8')
vitb16 = torch.hub.load('facebookresearch/dino', 'dino_vitb16') vitb16 = torch.hub.load('facebookresearch/dino:main', 'dino_vitb16')
vitb8 = torch.hub.load('facebookresearch/dino', 'dino_vitb8') vitb8 = torch.hub.load('facebookresearch/dino:main', 'dino_vitb8')
resnet50 = torch.hub.load('facebookresearch/dino', 'dino_resnet50') resnet50 = torch.hub.load('facebookresearch/dino:main', 'dino_resnet50')
``` ```
## Training ## Training
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment