ray.rllib.core.rl_模块.多重_rl_模块.多重RL模块.遍历每个模块#

MultiRLModule.foreach_module(func: Callable[[str, RLModule, Any | None], T], *, return_dict: bool = False, **kwargs) List[T] | Dict[str, T][源代码]#

使用每个 (module_id, module) 调用给定的函数。

参数:
  • func – 每个 (module_id, module) 元组调用的函数。

  • return_dict – 是否返回一个字典,该字典将 ModuleID 映射到调用 func 的各个模块的返回值。如果为 False(默认),则返回一个列表。

返回:

所有调用 func([module_id, module, **kwargs]) 的返回值列表,或者是一个字典(如果 return_dict=True),将 ModuleIDs 映射到各自模型的返回值。