Manager

Manager#

class stable_ssl.manager.Manager(*args, **kwargs)[source]#

Bases: Checkpointable

Manages training with logging, scheduling, and checkpointing support.

Parameters:

  • trainer (Union[dict, DictConfig, pl.Trainer]) – PyTorch Lightning trainer configuration or instance.

  • module (Union[dict, DictConfig, pl.LightningModule]) – Lightning module configuration or instance.

  • data (Union[dict, DictConfig, pl.LightningDataModule]) – Data module configuration or instance.

  • seed (int, optional) – Random seed for reproducibility. Defaults to None.

  • ckpt_path (str, optional) – Path to checkpoint for resuming training. Defaults to None.

checkpoint(*args, **kwargs)[source]#

Resubmits the same callable with the same arguments

Contents