code: def parse_args():
      parser = argparse.ArgumentParser()
      parser.add_argument(
          "--pop-size",
          type=int,
          default=128,
          help="ES population size.",
      )
      parser.add_argument(
          "--hidden-size",
          type=int,
          default=20,
          help="Policy hidden size.",
      )
      parser.add_argument(
          "--num-tests",
          type=int,
          default=100,
          help="Number of test rollouts.",
      )
      parser.add_argument(
          "--n-repeats",
          type=int,
          default=16,
          help="Training repetitions.",
      )
      parser.add_argument(
          "--max-iter",
          type=int,
          default=500,
          help="Max training iterations.",
      )
      parser.add_argument(
          "--test-interval", type=int, default=50, help="Test interval."
      )
      parser.add_argument(
          "--log-interval",
          type=int,
          default=10,
          help="Logging interval.",
      )
      parser.add_argument(
          "--seed",
          type=int,
          default=123,
          help="Random seed for training.",
      )
      parser.add_argument(
          "--init-std", type=float, default=0.5, help="Initial std."
      )
      parser.add_argument("--gpu-id", type=str, help="GPU(s) to use.")
      parser.add_argument(
          "--debug", action="store_true", help="Debug mode."
      )
      config, _ = parser.parse_known_args()
      return config
  
file_path: evojax/examples/train_slimevolley.py
imports: import argparse
  import os
  import shutil
  import jax
  from evojax.task.slimevolley import SlimeVolley
  from evojax.policy.mlp import MLPPolicy
  from evojax.algo import CMA
  from evojax import Trainer
  from evojax import util
label_desc: Use image node_id 1 for calling a global function with example usage: parse_args() and returns: config