after_exception#

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

Called after one or more workers have raised an exception.

Parameters:
  • run_context (TrainRunContext) – The TrainRunContext for the current training run.

  • worker_exceptions (Dict[int, Exception]) – A dict from worker world rank to the exception raised by that worker.