HardwareConfig

Contents

HardwareConfig#

class stable_ssl.config.HardwareConfig(seed: int | None = None, float16: bool = False, world_size: int = 1, device: str = 'cuda')[source]#

Bases: object

Configuration for the hardware parameters.

Parameters:

  • seed (int, optional) – Random seed for reproducibility. Default is None.

  • float16 (bool, optional) – Whether to use mixed precision (float16) for training. Default is False.

  • world_size (int, optional) – Number of processes participating in distributed training. Default is 1.

  • device (str, optional) – The device to use for training. Default is “cuda” if available, else “cpu”.

Contents