Finegrain/refiners
1
0
Fork 0
mirror of https://github.com/finegrain-ai/refiners.git synced 2024-11-21 21:58:47 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix typing issue in dinov2 test
Some checks failed
CI / lint_and_typecheck (push) Has been cancelled
Details
Deploy docs to GitHub Pages / Deploy docs (push) Has been cancelled
Details
Spell checker / Spell check (push) Has been cancelled
Details

Browse source
This commit is contained in:
Pierre Chapuis 2024-09-18 10:41:25 +02:00
parent 336253f26b
commit 73089a4e2d
1 changed files with 3 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
tests/foundationals/dinov2/test_dinov2.py
View file

@ -60,13 +60,14 @@ def ref_model(
if "reg" not in flavor: if "reg" not in flavor:
kwargs["interpolate_offset"] = 0.0 kwargs["interpolate_offset"] = 0.0
model = torch.hub.load( # type: ignore model: torch.nn.Module = torch.hub.load( # type: ignore
model=flavor, model=flavor,
repo_or_dir=str(dinov2_repo_path), repo_or_dir=str(dinov2_repo_path),
source="local", source="local",
pretrained=False, # to turn off automatic weights download (see load_state_dict below) pretrained=False, # to turn off automatic weights download (see load_state_dict below)
**kwargs, **kwargs,
).to(device=test_device) )
model = model.to(device=test_device)
flavor = flavor.replace("_reg", "_reg4") flavor = flavor.replace("_reg", "_reg4")
weights = test_weights_path / f"{flavor}_pretrain.pth" weights = test_weights_path / f"{flavor}_pretrain.pth"