plismbench.models.lunit module
Models from Lunit company.
-
class plismbench.models.lunit.LunitViTS8(device: int | list[int] | None = -1, mixed_precision: bool = False)[source]
Bases: Extractor
ViT-S/8 from Lunit available at (1).
- Parameters:
device (int | list[int] | None = DEFAULT_DEVICE,) – Compute resources to use.
If None, will use all available GPUs.
If -1, extraction will run on CPU.
mixed_precision (bool = True) – Whether to use mixed_precision.
-
property transform: Compose
Transform method to apply element wise.