ray.rllib.core.rl_module.default_model_config.DefaultModelConfig.lstm_kernel_initializer

DefaultModelConfig.lstm_kernel_initializer: str | Callable | None = None

Initializer function or class descriptor for the weight/kernel matrices in the LSTM layer. 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.