Actually, the depth of the annotations' one_hot matrix should be num_classes.
Actually, the depth of the annotations' one_hot matrix should be num_classes.