causica.lightning.cli¶

Module Contents¶

Classes¶

class causica.lightning.cli.LightningCLIWithDefaults(model_class: type[pytorch_lightning.LightningModule] | collections.abc.Callable[Ellipsis, pytorch_lightning.LightningModule] | None = None, datamodule_class: type[pytorch_lightning.LightningDataModule] | collections.abc.Callable[Ellipsis, pytorch_lightning.LightningDataModule] | None = None, save_config_callback: type[SaveConfigCallback] | None = SaveConfigCallback, save_config_kwargs: dict[str, Any] | None = None, trainer_class: type[pytorch_lightning.Trainer] | collections.abc.Callable[Ellipsis, pytorch_lightning.Trainer] = Trainer, trainer_defaults: dict[str, Any] | None = None, seed_everything_default: bool | int = True, parser_kwargs: dict[str, Any] | dict[str, dict[str, Any]] | None = None, subclass_mode_model: bool = False, subclass_mode_data: bool = False, args: ArgsType = None, run: bool = True, auto_configure_optimizers: bool = True)[source]¶

Bases: pytorch_lightning.cli.LightningCLI

Implementation of a configurable command line tool for pytorch-lightning.

default_logger[source]¶

add_arguments_to_parser(parser)[source]¶

Implement to add extra arguments to the parser or link arguments.

Parameters:¶

parser¶

The parser object to which arguments can be added