LightningArgumentParser

class pytorch_lightning.cli.LightningArgumentParser(*args, **kwargs)

Bases: object

Extension of jsonargparse’s ArgumentParser for pytorch-lightning.

Initialize argument parser that supports configuration file input.

For full details of accepted arguments see ArgumentParser.__init__.

add_lightning_class_args(lightning_class, nested_key, subclass_mode=False, required=True)

Adds arguments from a lightning class to a nested key of the parser.

Parameters

• lightning_class (Union[Callable[..., Union[Trainer, LightningModule, LightningDataModule, Callback]], Type[Trainer], Type[LightningModule], Type[LightningDataModule], Type[Callback]]) – A callable or any subclass of {Trainer, LightningModule, LightningDataModule, Callback}.

• nested_key (str) – Name of the nested namespace to store arguments.

• subclass_mode (bool) – Whether allow any subclass of the given class.

• required (bool) – Whether the argument group is required.

Return type

List[str]

Returns

A list with the names of the class arguments added.

add_lr_scheduler_args(lr_scheduler_class=(<class 'torch.optim.lr_scheduler._LRScheduler'>, <class 'pytorch_lightning.cli.ReduceLROnPlateau'>), nested_key='lr_scheduler', link_to='AUTOMATIC')

Adds arguments from a learning rate scheduler class to a nested key of the parser.

Parameters

• lr_scheduler_class (Union[Type[_LRScheduler], Type[ReduceLROnPlateau], Tuple[Union[Type[_LRScheduler], Type[ReduceLROnPlateau]], ...]]) – Any subclass of torch.optim.lr_scheduler.{_LRScheduler, ReduceLROnPlateau}. Use tuple to allow subclasses.

• nested_key (str) – Name of the nested namespace to store arguments.

• link_to (str) – Dot notation of a parser key to set arguments or AUTOMATIC.

Return type

None

add_optimizer_args(optimizer_class=(<class 'torch.optim.optimizer.Optimizer'>, ), nested_key='optimizer', link_to='AUTOMATIC')

Adds arguments from an optimizer class to a nested key of the parser.

Parameters

• optimizer_class (Union[Type[Optimizer], Tuple[Type[Optimizer], ...]]) – Any subclass of torch.optim.Optimizer. Use tuple to allow subclasses.

• nested_key (str) – Name of the nested namespace to store arguments.

• link_to (str) – Dot notation of a parser key to set arguments or AUTOMATIC.

Return type

None