utils.py

"""
@author: Yanzuo Lu
@author: oliveryanzuolu@gmail.com
"""

import logging

logger = logging.getLogger()


class AverageMeter:
    def __init__(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