diff --git a/baselines/deepq/models.py b/baselines/deepq/models.py index e2ac79c..803d104 100644 --- a/baselines/deepq/models.py +++ b/baselines/deepq/models.py @@ -2,9 +2,9 @@ import tensorflow as tf import tensorflow.contrib.layers as layers -def _mlp(hiddens, inpt, num_actions, scope, reuse=False, layer_norm=False): +def _mlp(hiddens, input_, num_actions, scope, reuse=False, layer_norm=False): with tf.variable_scope(scope, reuse=reuse): - out = inpt + out = input_ for hidden in hiddens: out = layers.fully_connected(out, num_outputs=hidden, activation_fn=None) if layer_norm: @@ -30,9 +30,9 @@ def mlp(hiddens=[], layer_norm=False): return lambda *args, **kwargs: _mlp(hiddens, layer_norm=layer_norm, *args, **kwargs) -def _cnn_to_mlp(convs, hiddens, dueling, inpt, num_actions, scope, reuse=False, layer_norm=False): +def _cnn_to_mlp(convs, hiddens, dueling, input_, num_actions, scope, reuse=False, layer_norm=False): with tf.variable_scope(scope, reuse=reuse): - out = inpt + out = input_ with tf.variable_scope("convnet"): for num_outputs, kernel_size, stride in convs: out = layers.convolution2d(out,