from utils.vis_helper import *
from archs import *

# graphviz_draw_darts(c100_50etr_beambest[-1])
graphviz_draw_darts(Genotype(normal=[('skip_connect', 0), ('skip_connect', 1), ('skip_connect', 1), ('dil_conv_3x3', 2), ('max_pool_3x3', 1), ('dil_conv_3x3', 2), ('avg_pool_3x3', 0), ('sep_conv_3x3', 1)], normal_concat=[2, 3, 4, 5], reduce=[('dil_conv_5x5', 0), ('dil_conv_5x5', 1), ('sep_conv_5x5', 1), ('sep_conv_5x5', 2), ('max_pool_3x3', 1), ('avg_pool_3x3', 2), ('max_pool_3x3', 0), ('max_pool_3x3', 1)], reduce_concat=[2, 3, 4, 5]))
# for i in range(8):
  # graphviz_draw_darts(c100_50etr_uniform_top8[i], label=f"c100 {i}")
