def inference(images, labels, is_training_dropout, is_training_bn, config): embds, end_points = get_embd(images, is_training_dropout, is_training_bn, config) logits = get_logits(embds, labels, config) end_points['logits'] = logits return embds, logits, end_points
def inference(images, labels, is_training_dropout, is_training_bn, config): embds = get_embd(images, is_training_dropout, is_training_bn) logits = get_logits(embds, labels, config) return embds, logits