lightning_template.utils.callbacks.save_and_log_config_callback

Contents

`lightning_template.utils.callbacks.save_and_log_config_callback`#

Module Contents#

Classes#

SaveAndLogConfigCallback

Saves and logs a LightningCLI config to the log_dir when training starts.

class lightning_template.utils.callbacks.save_and_log_config_callback.SaveAndLogConfigCallback(parser: LightningArgumentParser, config: jsonargparse.Namespace, config_filename: str = 'config.yaml', overwrite: bool = False, multifile: bool = False, save_to_log_dir: bool = True)#

Bases: lightning.pytorch.cli.SaveConfigCallback

Saves and logs a LightningCLI config to the log_dir when training starts.

setup(trainer: lightning.pytorch.Trainer, pl_module: lightning.pytorch.LightningModule, stage: str | None = None) → None#: Called when fit, validate, test, predict, or tune begins.