ray.rllib.core.learner.learner.Learner.get_parameters

abstract Learner.get_parameters(module: RLModule) → Sequence[torch.Tensor | tf.Variable]

Returns the list of parameters of a module.

This should be overriden in framework specific learner. For example in torch it will return .parameters(), while in tf it returns .trainable_variables.

Parameters:

module – The module to get the parameters from.

Returns:

The parameters of the module.