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
当模型训练时采用的参数更新的方式,
采用自定义的AverageMeter类来管理一些变量的更新。
读取某个变量的时候,通过对象.属性的方式来读取
在初始化的时候就调用的重置方法reset。
当需要进行变量更新时,调用该类对象的update方法,是对所有的batchsize的损失取平均