hz2475 commited on
Commit
9e9b958
·
verified ·
1 Parent(s): 5697877

Create README.md

Browse files
Files changed (1) hide show
  1. README.md +16 -0
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
+ ```