Spaces:
Runtime error
Runtime error
File size: 554 Bytes
bc32eea |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 |
class AverageMeter(object):
"""Computes and stores the average and current value.
Adapted from: https://github.com/pytorch/examples/blob/master/imagenet/train.py
"""
def __init__(self):
self.val = 0.
self.avg = 0.
self.sum = 0.
self.count = 0.
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
|