Python MLP Exemples

Exemple #1

 def q(obs):
     flatten = lambda x: jnp.reshape(x, (-1, ))
     network = hk.Sequential(
         [flatten, nets.MLP([FLAGS.hidden_units, num_actions])])
     return network(obs)

Exemple #2

Fichier : simple_dqn.py Projet : deepmind/rlax

 def q(obs):
     network = hk.Sequential(
         [hk.Flatten(),
          nets.MLP([FLAGS.hidden_units, num_actions])])
     return network(obs)