lightning_template.utils.cli.trainer

Module Contents

Classes

Trainer

Functions

_use_auto_lr_schedule(old_func)

lightning_template.utils.cli.trainer._use_auto_lr_schedule(old_func)

class lightning_template.utils.cli.trainer.Trainer(num_folds: int | None = None, *args, **kwargs)

Bases: lightning.pytorch.Trainer