ray.rllib.algorithms.algorithm_config.AlgorithmConfig.callbacks

AlgorithmConfig.callbacks(callbacks_class) → AlgorithmConfig

Sets the callbacks configuration.

Parameters:

callbacks_class – Callbacks class, whose methods will be run during various phases of training and environment sample collection. See the DefaultCallbacks class and examples/custom_metrics_and_callbacks.py for more usage information.

Returns:

This updated AlgorithmConfig object.