arlbench.core.algorithms.dqn.models

Q-Networks for DQN.

Classes

CNNQ(action_dim[, activation, hidden_size, ...])	A CNN-based Q-Network for DQN.
MLPQ(action_dim[, activation, hidden_size, ...])	An MLP-based Q-Network for DQN.

class arlbench.core.algorithms.dqn.models.CNNQ(action_dim, activation='tanh', hidden_size=512, discrete=True, parent=<flax.linen.module._Sentinel object>, name=None)

Bases: Module

A CNN-based Q-Network for DQN.

__call__(x)

Applies the CNN to the input.

setup()

Initializes the CNN Q-Network.

class arlbench.core.algorithms.dqn.models.MLPQ(action_dim, activation='tanh', hidden_size=64, discrete=True, parent=<flax.linen.module._Sentinel object>, name=None)

Bases: Module

An MLP-based Q-Network for DQN.

__call__(x)

Applies the MLP to the input.

setup()

Initializes the MLP Q-Network.

previous

arlbench.core.algorithms.dqn.dqn

next

arlbench.core.algorithms.ppo