ray.rllib.core.rl_module.default_model_config.DefaultModelConfig.conv_kernel_initializer

DefaultModelConfig.conv_kernel_initializer: str | Callable | None = None

Initializer function or class descriptor for the weight/kernel matrices in the stack configured by conv_filters. Supported values are the initializer names (str), classes or functions listed by the frameworks (torch). See https://pytorch.org/docs/stable/nn.init.html for torch. If None (default), the default initializer defined by torch is used.