utils/third_party/nni_new/nas/benchmarks/nasbench101/constants.py

INPUT = 'input'
OUTPUT = 'output'
CONV3X3_BN_RELU = 'conv3x3-bn-relu'
CONV1X1_BN_RELU = 'conv1x1-bn-relu'
MAXPOOL3X3 = 'maxpool3x3'


LABEL2ID = {
    INPUT: -1,
    OUTPUT: -2,
    CONV3X3_BN_RELU: 0,
    CONV1X1_BN_RELU: 1,
    MAXPOOL3X3: 2
}