ray.train.lightning.RayTrainReportCallback#

class ray.train.lightning.RayTrainReportCallback(*args: Any, **kwargs: Any)[source]#

Bases: pytorch_lightning.callbacks.Callback

A simple callback that reports checkpoints to Ray on train epoch end.

PublicAPI (beta): This API is in beta and may change before becoming stable.

Methods