ConfigurableBaseEstimator.trainer

ConfigurableBaseEstimator.trainer(**kwargs)

Returns the trainer as configured by the estimator. Typically, this method is only called by functions in the estimator.

Parameters

kwargs (Any) -- Additional arguments that override the trainer arguments registered in the initializer of the estimator.

Return type

Trainer

Returns

A fully initialized PyTorch Lightning trainer.

Note

This function should be preferred over initializing the trainer directly. It ensures that the returned trainer correctly deals with LightKit components that may be introduced in the future.