ray.train.UserCallback

class ray.train.UserCallback

Bases: RayTrainCallback

Callback interface for custom user-defined callbacks to handling events during training.

This callback is called on the Ray Train controller process, not on the worker processes.

DeveloperAPI: This API may change across minor Ray releases.

Methods

__init__
after_exception	Called after one or more workers have raised an exception.
after_report	Called after all workers have reported a metric + checkpoint via ray.train.report.