Skip to content

Commit 418b0d8

Browse files
committed
Update README for torch hub
1 parent 07d713f commit 418b0d8

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

README.md

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -21,6 +21,14 @@ It has been tested on Pytorch 2.1.0 with CUDA 12.1 and Xformers. Create a ready
2121
conda env create -f environment.yml
2222
```
2323

24+
To quickly test and use our model, you can use Torch Hub:
25+
```python
26+
import torch
27+
model = torch.hub.load("serizba/salad", "dinov2_salad")
28+
model.eval()
29+
model.cuda()
30+
```
31+
2432
## Dataset
2533

2634
For training, download [GSV-Cities](https://github.com/amaralibey/gsv-cities) dataset. For evaluation download the desired datasets ([MSLS](https://github.com/FrederikWarburg/mapillary_sls), [NordLand](https://surfdrive.surf.nl/files/index.php/s/sbZRXzYe3l0v67W), [SPED](https://surfdrive.surf.nl/files/index.php/s/sbZRXzYe3l0v67W), or [Pittsburgh](https://data.ciirc.cvut.cz/public/projects/2015netVLAD/Pittsburgh250k/))

0 commit comments

Comments
 (0)