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 }