def _get_input_vars(self):
    return [InputVar(tf.float32, [None, 224, 224, 3], 'input'), InputVar(tf.int32, [None], 'label')]
def _get_input_vars(self):
    return [InputVar(tf.float32, [None, 224, 224, 3], 'input'), InputVar(tf.int32, [None], 'label')]