utils.py

import time
from datetime import datetime


class AverageMeter(object):
    """From https://github.com/pytorch/examples/blob/master/imagenet/main.py"""

    def __init__(self):
        self.val = 0
        self.avg = 0
        self.sum = 0
        self.count = 0

    def __repr__(self):
        return f'{self.avg:.2e}'

    def update(self, val, n=1):
        self.val = val
        self.sum += val * n
        self.count += n
        self.avg = self.sum / self.count


class dotdict(dict):
    def __getattr__(self, name):
        return self[name]


def get_readable_time():
    timestamp = time.time()
    normal_time = str(datetime.fromtimestamp(timestamp))
    temp = normal_time.split()
    first = temp[0].replace('-', '_')
    second = temp[1].split(':')
    second[-1] = second[-1].split('.')[0]
    second = "_".join(second)
    ret = first + "_" + second
    return ret