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

abstract Learner.get_parameters(module: ray.rllib.core.rl_module.rl_module.RLModule) → Sequence[Union[torch.Tensor, tensorflow.python.ops.variables.Variable]][source]#

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.