ray.rllib.algorithms.algorithm_config.AlgorithmConfig.callbacks

AlgorithmConfig.callbacks(callbacks_class) → AlgorithmConfig

Sets the callbacks configuration.

Parameters:

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

Returns:

This updated AlgorithmConfig object.