Unverified Commit 6d6f9524 authored by Chi Song's avatar Chi Song Committed by GitHub
Browse files

pdarts update (#1753)

parent 77e91e8b
from collections import OrderedDict
_counter = 0
def global_mutable_counting():
global _counter
_counter += 1
return _counter
class AverageMeterGroup(object):
def __init__(self):
self.meters = OrderedDict()
def update(self, data):
for k, v in data.items():
if k not in self.meters:
self.meters[k] = AverageMeter(k, ":4f")
self.meters[k].update(v)
def __str__(self):
return " ".join(str(v) for _, v in self.meters.items())
class AverageMeter(object):
"""Computes and stores the average and current value"""
def __init__(self, name, fmt=':f'):
self.name = name
self.fmt = fmt
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
def __str__(self):
fmtstr = '{name} {val' + self.fmt + '} ({avg' + self.fmt + '})'
return fmtstr.format(**self.__dict__)
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment