mrkwatkins.oakemu_rl.framework.ppo.PPOModel

Classes and functions

• PPOActionBundle
• PPOModel

Constants

ACTIVATIONS

ACTIVATIONS: dict[ActivationName, type[nn.Module]] = {'relu': nn.ReLU, 'leaky_relu': nn.LeakyReLU, 'elu': nn.ELU, 'gelu': nn.GELU, 'silu': nn.SiLU, 'tanh': nn.Tanh, 'sigmoid': nn.Sigmoid}

Activation modules by configuration name. Keep in step with ActivationName.

Source