Create README.md
Browse files
README.md
ADDED
|
@@ -0,0 +1,16 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
license: mit
|
| 3 |
+
---
|
| 4 |
+
# Dreamsim local checkpoints
|
| 5 |
+
all checkpoints needed to run dreamsim locally, verified on commit [2b3e78e](https://github.com/ssundaram21/dreamsim/commit/2b3e78ee4c104257bbed4a4c9084d4b30fe1665c)
|
| 6 |
+
|
| 7 |
+
## download
|
| 8 |
+
use git to download files to local
|
| 9 |
+
|
| 10 |
+
## run
|
| 11 |
+
|
| 12 |
+
```python
|
| 13 |
+
from dreamsim import dreamsim
|
| 14 |
+
model, preprocess = dreamsim(pretrained=True, device=self.device, cache_dir="/path/to/dreamsim_ckpts")
|
| 15 |
+
dist = model(tensor_a, tensor_b).item()
|
| 16 |
+
```
|