code: def make_networks(
      action_spec,
      policy_layer_sizes=(10, 10),
      critic_layer_sizes=(10, 10),
  ):
      """Creates networks used by the agent."""
  
      num_dimensions = np.prod(action_spec.shape, dtype=int)
      critic_layer_sizes = list(critic_layer_sizes) + [1]
  
      policy_network = snt.Sequential(
          [
              networks.LayerNormMLP(policy_layer_sizes),
              networks.MultivariateNormalDiagHead(num_dimensions),
          ]
      )
      critic_network = networks.CriticMultiplexer(
          critic_network=networks.LayerNormMLP(critic_layer_sizes)
      )
  
      return {
          "policy": policy_network,
          "critic": critic_network,
      }
  
file_path: acme/acme/agents/tf/mpo/agent_test.py
imports: import acme
  from acme import specs
  from acme.agents.tf import mpo
  from acme.testing import fakes
  from acme.tf import networks
  import numpy
  import sonnet
  from absl.testing import absltest
label_desc: Use image node_id 1 for calling a global function with example usage: make_networks(action_spec, policy_layer_sizes, critic_layer_sizes) and returns: dict