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

Learner._check_registered_optimizer(optimizer: Union[torch.optim.optimizer.Optimizer, keras.optimizers.optimizer_experimental.optimizer.Optimizer], params: Sequence[Union[torch.Tensor, tensorflow.python.ops.variables.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.