Function conv

rl3/a2c/neural_network.py:11–14

(inputs, nf, ks, strides, gain=1.0)

def conv(inputs, nf, ks, strides, gain=1.0):
    return tf.layers.conv2d(inputs=inputs, filters=nf, kernel_size=ks,
                            strides=(strides, strides), activation=tf.nn.relu,
                            kernel_initializer=tf.orthogonal_initializer(gain=gain))


def dense(inputs, n, act=tf.nn.relu, gain=1.0):