Files
DALLE2-pytorch/dalle2_pytorch/utils.py
2022-05-22 15:39:38 -07:00

20 lines
342 B
Python

import time
# time helpers
class Timer:
def __init__(self):
self.reset()
def reset(self):
self.last_time = time.time()
def elapsed(self):
return time.time() - self.last_time
# print helpers
def print_ribbon(s, symbol = '=', repeat = 40):
flank = symbol * repeat
return f'{flank} {s} {flank}'