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.