from __future__ import absolute_import
from __future__ import division
from __future__ import print_function

import torch

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
 if self.count > 0:
 self.avg = self.sum / self.count