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

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

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.