External library integrations for Ray Tune#

PyTorch Lightning (tune.integration.pytorch_lightning)#

TuneReportCallback

TuneReportCheckpointCallback

PyTorch Lightning report and checkpoint callback

XGBoost (tune.integration.xgboost)#

TuneReportCallback

TuneReportCheckpointCallback

XGBoost callback to save checkpoints and report metrics.

LightGBM (tune.integration.lightgbm)#

TuneReportCallback

TuneReportCheckpointCallback

Creates a callback that reports metrics and checkpoints model.