code: def parse_args():
      parser = argparse.ArgumentParser()
      parser.add_argument(
          "--pop-size", type=int, default=64, help="NE population size."
      )
      parser.add_argument(
          "--batch-size",
          type=int,
          default=1024,
          help="Batch size for training.",
      )
      parser.add_argument(
          "--max-iter",
          type=int,
          default=5000,
          help="Max training iterations.",
      )
      parser.add_argument(
          "--test-interval",
          type=int,
          default=1000,
          help="Test interval.",
      )
      parser.add_argument(
          "--log-interval",
          type=int,
          default=100,
          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.006,
          help="Center learning rate.",
      )
      parser.add_argument(
          "--std-lr",
          type=float,
          default=0.089,
          help="Std learning rate.",
      )
      parser.add_argument(
          "--init-std", type=float, default=0.039, 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_mnist.py
imports: import argparse
  import os
  import shutil
  from evojax import Trainer
  from evojax.task.mnist import MNIST
  from evojax.policy.convnet import ConvNetPolicy
  from evojax.algo import PGPE
  from evojax import util
label_desc: Use image node_id 1 for calling a global function with example usage: parse_args() and returns: config