HardwareConfig

Contents

HardwareConfig#

class stable_ssl.config.HardwareConfig(seed: int | None = None, float16: bool = False, gpu: int = 0, world_size: int = 1, port: int | None = None)[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.

  • gpu (int, optional) – GPU device ID to use for training. Default is 0.

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

  • port (int, optional) – Port number for distributed training. Default is None.

Contents