code: def parse_arch_str(arch_str):
      mp = {
          "none": NONE,
          "skip_connect": SKIP_CONNECT,
          "nor_conv_1x1": CONV_1X1,
          "nor_conv_3x3": CONV_3X3,
          "avg_pool_3x3": AVG_POOL_3X3,
      }
      m: re.Match = re.match(r"\|(.*)~0\|\+\|(.*)~0\|(.*)~1\|\+\|(.*)~0\|(.*)~1\|(.*)~2\|", arch_str)  # type: ignore
      return {
          "0_1": mp[m.group(1)],
          "0_2": mp[m.group(2)],
          "1_2": mp[m.group(3)],
          "0_3": mp[m.group(4)],
          "1_3": mp[m.group(5)],
          "2_3": mp[m.group(6)],
      }
  
file_path: nni/nni/nas/benchmark/nasbench201/db_gen.py
imports: import argparse
  import re
  import tqdm
  import torch
  from nni.nas.benchmark.utils import load_benchmark
  from .constants import NONE, SKIP_CONNECT, CONV_1X1, CONV_3X3, AVG_POOL_3X3
  from .schema import Nb201TrialConfig, Nb201TrialStats, Nb201IntermediateStats
label_desc: Use image node_id 1 for calling a global function with example usage: parse_arch_str(arch_str) and returns: dict