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]]

没有可翻译的文本。

state_forward: Callable

没有可翻译的文本。

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 模型状态前向函数。