ray.train.context.get_context

ray.train.context.get_context() → TrainContext

Get or create a singleton training context.

The context is only available within a function passed to Ray Train.

See the TrainContext API reference to see available methods.