dreamsim_ckpts / README.md
hz2475's picture
Create README.md
9e9b958 verified
metadata
license: mit

Dreamsim local checkpoints

all checkpoints needed to run dreamsim locally, verified on commit 2b3e78e

download

use git to download files to local

run

from dreamsim import dreamsim
model, preprocess = dreamsim(pretrained=True, device=self.device, cache_dir="/path/to/dreamsim_ckpts")
dist = model(tensor_a, tensor_b).item()