ray.rllib.algorithms.algorithm_config.AlgorithmConfig.callbacks

AlgorithmConfig.callbacks(callbacks_class) → ray.rllib.algorithms.algorithm_config.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.