ray.train.lightning.RayTrainReportCallback

class ray.train.lightning.RayTrainReportCallback(*args: Any, **kwargs: Any)

Bases: RayTrainReportCallback

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_metrics and reports together with the checkpoint on each training epoch end.

Checkpoints will be saved in the following structure:

checkpoint_{timestamp}/ Ray Train’s checkpoint folder └─ checkpoint.ckpt Lightning’s checkpoint format

For customized reporting and checkpointing logic, implement your own lightning.pytorch.callbacks.Callback following this user guide: Saving and Loading Checkpoints.

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

Methods

Attributes

CHECKPOINT_NAME