AverageMeter一个用来记录和更新变量的工具
之前使用过很多次,也没有做很好的记录,这次记录下来。AverageMeter可以记录当前的输出,累加到某个变量之中,然后根据需要可以打印出历史上的平均
这个class是一种常用的工具
该class内容如下
class AverageMeter(object):
def __init__(self):
self.reset()
def is_empty(self):
return self.cnt == 0
def reset(self):
self.avg = 0.
self.sum = 0.
self.cnt = 0
def update(self, val, n=1):
self.sum += val*n
self.cnt += n
self.avg = self.sum / self.cnt
这样在调用的时候,可以先声明
obj = AverageMeter()
需要将所有变量清零的时候,调用
obj.reset()
然后需要更新某个变量的时候
obj.update(x)
这样的话比如想要求平均,就可以直接用
obj.avg
即可
posted on 2020-12-13 17:03 YongjieShi 阅读(3553) 评论(0) 编辑 收藏 举报