lightning_template.tools.model.batch_size_finder

Classes

LightningCLI

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

Functions

main()

Module Contents

class lightning_template.tools.model.batch_size_finder.LightningCLI(save_config_callback: Type[lightning.pytorch.cli.SaveConfigCallback] | None = SaveAndLogConfigCallback, trainer_class: Type[lightning_template.utils.cli.trainer._Trainer] | Callable[Ellipsis, lightning_template.utils.cli.trainer._Trainer] = Trainer, *args, **kwargs)

Bases: lightning_template.utils.LightningCLI

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

before_instantiate_classes() → None

Implement to run some code before instantiating the classes.

lightning_template.tools.model.batch_size_finder.main()