"""Implements Average Meter."""
[docs]class AverageMeter:
"""Computes and stores the average and current value."""
def __init__(self):
"""Constructor class for Average Meter."""
self.val = 0
self.avg = 0
self.sum = 0
self.count = 0
[docs] def reset(self):
"""Method to reset all the internal variables."""
self.val = 0
self.avg = 0
self.sum = 0
self.count = 0
[docs] def update(self, val, n=1):
"""Method to update values.
Args:
val: The current values.
n: The count.
"""
self.val = val
self.sum += val * n
self.count += n
self.avg = self.sum / self.count