`evox.problems.neuroevolution.utils`¶

模块内容¶

类¶

ModelStateForwardResult

函数¶

get_vmap_model_state_forward

获取用于 vmap 和非 vmap 模型的模型状态前向函数。当 get_non_vmap 为 False 时，该函数仅返回 vmap 模型状态前向函数。当 get_non_vmap 为 True 时，该函数返回 vmap 和非 vmap 模型状态前向函数。

API¶

class evox.problems.neuroevolution.utils.ModelStateForwardResult[源代码]¶

Bases: typing.NamedTuple

init_state: Tuple[Dict[str, torch.Tensor], Dict[str, torch.Tensor]]¶: None

state_forward: Callable¶: None

evox.problems.neuroevolution.utils.get_vmap_model_state_forward(model: torch.nn.Module, pop_size: int, device: torch.device, in_dims=(0, None), randomness='different') → evox.problems.neuroevolution.utils.ModelStateForwardResult[源代码]¶: 获取用于 vmap 和非 vmap 模型的模型状态前向函数。当 get_non_vmap 为 False 时，该函数仅返回 vmap 模型状态前向函数。当 get_non_vmap 为 True 时，该函数返回 vmap 和非 vmap 模型状态前向函数。