Skip to content

classification

dino_vitb8()

Yield a dino_vitb8 model.

Source code in quadra/utils/tests/fixtures/models/classification.py 
36
37
38
39
40
41
42
43
44
45
@pytest.fixture
def dino_vitb8():
    """Yield a dino_vitb8 model."""
    yield TorchHubNetworkBuilder(
        repo_or_dir="facebookresearch/dino:main",
        model_name="dino_vitb8",
        pretrained=False,
        freeze=True,
        exportable=True,
    )

dino_vits8()

Yield a dino_vits8 model.

Source code in quadra/utils/tests/fixtures/models/classification.py 
24
25
26
27
28
29
30
31
32
33
@pytest.fixture
def dino_vits8():
    """Yield a dino_vits8 model."""
    yield TorchHubNetworkBuilder(
        repo_or_dir="facebookresearch/dino:main",
        model_name="dino_vits8",
        pretrained=False,
        freeze=True,
        exportable=True,
    )

resnet18()

Yield a resnet18 model.

Source code in quadra/utils/tests/fixtures/models/classification.py 
6
7
8
9
@pytest.fixture
def resnet18():
    """Yield a resnet18 model."""
    yield TimmNetworkBuilder("resnet18", pretrained=False, freeze=True, exportable=True)

resnet50()

Yield a resnet50 model.

Source code in quadra/utils/tests/fixtures/models/classification.py 
12
13
14
15
@pytest.fixture
def resnet50():
    """Yield a resnet50 model."""
    yield TimmNetworkBuilder("resnet50", pretrained=False, freeze=True, exportable=True)

vit_tiny_patch16_224()

Yield a vit_tiny_patch16_224 model.

Source code in quadra/utils/tests/fixtures/models/classification.py 
18
19
20
21
@pytest.fixture
def vit_tiny_patch16_224():
    """Yield a vit_tiny_patch16_224 model."""
    yield TimmNetworkBuilder("vit_tiny_patch16_224", pretrained=False, freeze=True, exportable=True)