ray.rllib.models.modelv2.ModelV2.可训练变量

ModelV2.trainable_variables(as_dict: bool = False) → List[numpy.array | jnp.ndarray | tf.Tensor | torch.Tensor] | Dict[str, numpy.array | jnp.ndarray | tf.Tensor | torch.Tensor]

返回此模型的可训练变量列表。

参数:

as_dict – 变量是否应作为字典值返回（使用描述性键）。

返回:

此 ModelV2 的所有可训练（tf）/requires_grad（torch）变量的列表（如果 as_dict 为 True，则为字典）。