| import json |
| import time |
|
|
|
|
| class TimeCounter: |
| def __init__(self) -> None: |
| pass |
|
|
| def clear(self): |
| self.timedict = {} |
| self.basetime = time.perf_counter() |
|
|
| def timeit(self, name): |
| nowtime = time.perf_counter() - self.basetime |
| self.timedict[name] = nowtime |
| self.basetime = time.perf_counter() |
|
|
|
|
| class TimeHolder: |
| def __init__(self) -> None: |
| self.timedict = {} |
|
|
| def update(self, _timedict: dict): |
| for k, v in _timedict.items(): |
| if k not in self.timedict: |
| self.timedict[k] = AverageMeter(name=k, val_only=True) |
| self.timedict[k].update(val=v) |
|
|
| def final_res(self): |
| return {k: v.avg for k, v in self.timedict.items()} |
|
|
| def __str__(self): |
| return json.dumps(self.final_res(), indent=2) |
|
|
|
|
| class AverageMeter(object): |
| """Computes and stores the average and current value""" |
|
|
| def __init__(self, name, fmt=":f", val_only=False): |
| self.name = name |
| self.fmt = fmt |
| self.val_only = val_only |
| 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): |
| if self.val_only: |
| fmtstr = "{name} {val" + self.fmt + "}" |
| else: |
| fmtstr = "{name} {val" + self.fmt + "} ({avg" + self.fmt + "})" |
| return fmtstr.format(**self.__dict__) |
|
|
