src.utils package

Module contents

src.utils.extras(config: DictConfig) → None

Applies optional utilities, controlled by config flags.

Utilities: - Ignoring python warnings - Rich config printing

src.utils.finish(config: DictConfig, model: LightningModule, datamodule: LightningDataModule, trainer: Trainer, callbacks: List[Callback], logger: List[LightningLoggerBase]) → None

Makes sure everything closed properly.

src.utils.get_logger(name='src.utils') → Logger

Initializes multi-GPU-friendly python command line logger.

src.utils.log_hyperparameters(config: DictConfig, model: LightningModule, datamodule: LightningDataModule, trainer: Trainer, callbacks: List[Callback], logger: List[LightningLoggerBase]) → None

Controls which config parts are saved by Lightning loggers.

Additionaly saves: - number of model parameters

src.utils.print_config(config: DictConfig, print_order: Sequence[str] = ('datamodule', 'model', 'callbacks', 'logger', 'trainer'), resolve: bool = True) → None

Prints content of DictConfig using Rich library and its tree structure.

Parameters:

• config (DictConfig) – Configuration composed by Hydra.

• print_order (Sequence[str], optional) – Determines in what order config components are printed.

• resolve (bool, optional) – Whether to resolve reference fields of DictConfig.