KMeans.trainer

KMeans.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.