reinforcement-learning icon indicating copy to clipboard operation
reinforcement-learning copied to clipboard

The output layer should not using RELU activation function.

Open wanjunhong0 opened this issue 5 years ago • 0 comments

tf.contrib.layers.fully_connected (activation_fn: Activation function. The default value is a ReLU function. Explicitly set it to None to skip it and maintain a linear activation.)

In the original paper, the last FC layer is set to linear layer. So the in the dpn.py, self.predictions = tf.contrib.layers.fully_connected(fc1, len(VALID_ACTIONS)) should be set to self.predictions = tf.contrib.layers.fully_connected(fc1, len(VALID_ACTIONS), activation_fn=None)

wanjunhong0 avatar Dec 24 '18 07:12 wanjunhong0