code: def main():
      task = Task.init(
          project_name="examples", task_name="MegEngine MNIST train"
      )  # noqa
  
      parser = argparse.ArgumentParser(
          description="MegEngine MNIST Example"
      )
      parser.add_argument(
          "--epoch",
          type=int,
          default=10,
          help="number of training epoch(default: 10)",
      )
      parser.add_argument(
          "--lr",
          type=float,
          default=0.01,
          help="learning rate(default: 0.01)",
      )
      parser.add_argument(
          "--momentum",
          type=float,
          default=0.9,
          help="SGD momentum (default: 0.9)",
      )
      parser.add_argument(
          "--wd",
          type=float,
          default=5e-4,
          help="SGD weight decay(default: 5e-4)",
      )
  
      args = parser.parse_args()
      dataloader = build_dataloader()
      train(dataloader, args)
  
file_path: clearml/examples/frameworks/megengine/megengine_mnist.py
imports: import argparse
  import os
  import sys
  from tempfile import gettempdir
  import numpy
  from clearml import Task
  from tensorboardX import SummaryWriter
label_desc: Use image node_id 3 for calling a global function with example usage: main() without return types