pytorch 平均メーター

平均的な金額

class AverageMeter(object):
    """Computes and stores the average and current value"""
    def __init__(self):
        self.reset()

    def reset(self):
        self.val = 0
        self.avg = 0
        self.sum = 0
        self.count = 0

    def update(self, val, n=1):
        self.val = val
        self.sum += val * n
        self.count += n
        self.avg = self.sum / self.count

呼び出し:
トレーニング、各エポックの前:
loss = AverageMeter()

イターごとの現在のエポック:

losses.update(to_python_float(reduced_loss.data), input.size(0))

# item() is a recent addition, so this helps with backward compatibility.
def to_python_float(t):
    if hasattr(t, 'item'):
        return t.item()
    else:
        return t[0]