Function get_model

examples/reinforcement_learning/tutorial_atari_pong.py:68–73

(inputs_shape)

# policy network
def get_model(inputs_shape):
    ni = tl.layers.Input(inputs_shape)
    nn = tl.layers.Dense(n_units=H, act=tf.nn.relu, name='hidden')(ni)
    nn = tl.layers.Dense(n_units=3, name='output')(nn)
    M = tl.models.Model(inputs=ni, outputs=nn, name="mlp")
    return M


model = get_model([None, D])