ray.train.UserCallback.after_exception#

UserCallback.after_exception(run_context: TrainRunContext, worker_exceptions: Dict[int, Exception])[source]#

Called after one or more workers have raised an exception.

Parameters:

  • run_context – The TrainRunContext for the current training run.

  • worker_exceptions – A dict from worker world rank to the exception raised by that worker.