Esempi in Python per TrainingContext.log

Esempio n. 1

File: cifar_cnce.py Progetto: mjendrusch/torchsupport

def valid_callback(args, ctx: TrainingContext = None):
    ctx.log(images=LogImage(args.sample))
    labels = args.prediction.argmax(dim=1)
    for idx in range(10):
        positive = args.sample[labels == idx]
        if positive.size(0) != 0:
            ctx.log(**{f"classified {idx}": LogImage(positive)})

Esempio n. 2

def valid_callback(args, ctx: TrainingContext = None):
    ctx.log(images=LogImage(args.condition))
    labels = args.distribution.logits.argmax(dim=1)
    for idx in range(10):
        positive = args.condition[labels == idx]
        if positive.size(0) != 0:
            ctx.log(**{f"classified {idx}": LogImage(positive)})