hive.agents.qnets.atari.nature_atari_dqn module

class hive.agents.qnets.atari.nature_atari_dqn.NatureAtariDQNModel(in_dim)

Bases: ConvNetwork

The convolutional network used to train the DQN agent in the original Nature paper: https://www.nature.com/articles/nature14236

Parameters

in_dim (tuple) – The tuple of observations dimension (channels, width, height).

training: bool