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.