예제 #1
0
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
예제 #2
0
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