Function Timer

rft/trainer/utils/__init__.py:29–34

(name: str)

@contextmanager
def Timer(name: str):
    start = time.perf_counter()
    yield
    end = time.perf_counter()
    # torch.cuda.synchronize()
    logger.info(f"[TIMER] {name}: {(end - start)*1000:.2f} ms")

logger = logging.getLogger("ARL")    
logger.setLevel(logging.INFO)