ray.rllib.core.learner.learner.Learner._check_registered_optimizer#

Learner._check_registered_optimizer(optimizer: torch.optim.Optimizer | keras.optimizers.Optimizer, params: Sequence[torch.Tensor | tf.Variable]) → None[source]#

Checks that the given optimizer and parameters are valid for the framework.

Parameters:

  • optimizer – The optimizer object to check.

  • params – The list of parameters to check.