External library integrations for Ray Tune#

PyTorch Lightning (tune.integration.pytorch_lightning)#

TuneReportCallback(*args, **kwargs)

TuneReportCheckpointCallback(*args, **kwargs)

PyTorch Lightning report and checkpoint callback

XGBoost (tune.integration.xgboost)#

TuneReportCallback(*args, **kwargs)

TuneReportCheckpointCallback(*args, **kwargs)

XGBoost callback to save checkpoints and report metrics.

LightGBM (tune.integration.lightgbm)#

TuneReportCallback([metrics, ...])

TuneReportCheckpointCallback([metrics, ...])

Creates a callback that reports metrics and checkpoints model.