- class ray.train.lightning.RayTrainReportCallback(*args: Any, **kwargs: Any)#
A simple callback that reports checkpoints to Ray on train epoch end.
This callback is a subclass of lightning.pytorch.callbacks.Callback.
It fetches the latest
trainer.callback_metricsand reports together with the checkpoint on each training epoch end.
Checkpoints will be saved in the following structure:
checkpoint_00000*/ Ray Train Checkpoint └─ checkpoint.ckpt PyTorch Lightning Checkpoint
PublicAPI (beta): This API is in beta and may change before becoming stable.