code: def parse_args():
      parser = argparse.ArgumentParser()
      parser.add_argument(
          "--pop-size", type=int, default=64, help="NE population size."
      )
      parser.add_argument(
          "--hidden-size",
          type=int,
          default=64,
          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=1000,
          help="Max training iterations.",
      )
      parser.add_argument(
          "--test-interval",
          type=int,
          default=100,
          help="Test interval.",
      )
      parser.add_argument(
          "--log-interval",
          type=int,
          default=20,
          help="Logging interval.",
      )
      parser.add_argument(
          "--seed",
          type=int,
          default=42,
          help="Random seed for training.",
      )
      parser.add_argument(
          "--center-lr",
          type=float,
          default=0.05,
          help="Center learning rate.",
      )
      parser.add_argument(
          "--std-lr", type=float, default=0.1, help="Std learning rate."
      )
      parser.add_argument(
          "--init-std", type=float, default=0.1, help="Initial std."
      )
      parser.add_argument("--gpu-id", type=str, help="GPU(s) to use.")
      parser.add_argument(
          "--easy", action="store_true", help="Easy mode."
      )
      parser.add_argument(
          "--pi",
          action="store_true",
          help="Permutation invariant policy.",
      )
      parser.add_argument(
          "--cma", action="store_true", help="Training with CMA-ES."
      )
      parser.add_argument(
          "--debug", action="store_true", help="Debug mode."
      )
      config, _ = parser.parse_known_args()
      return config
  
file_path: evojax/examples/train_cartpole.py
imports: import argparse
  import os
  import shutil
  import jax
  from evojax import Trainer
  from evojax.task.cartpole import CartPoleSwingUp
  from evojax.policy import MLPPolicy
  from evojax.policy import PermutationInvariantPolicy
  from evojax.algo import PGPE
  from evojax.algo import CMA
  from evojax import util
  from evojax.util import get_tensorboard_log_fn
label_desc: Use image node_id 1 for calling a global function with example usage: parse_args() and returns: config