ray.rllib.core.learner.learner.Learner.获取参数#

abstract Learner.get_parameters(module: RLModule) Sequence[torch.Tensor | tf.Variable][源代码]#

返回模块的参数列表。

这应该在特定框架的学习器中被重写。例如,在 torch 中它将返回 .parameters(),而在 tf 中它返回 .trainable_variables。

参数:

module – 获取参数的模块。

返回:

模块的参数。