mmdeploy/tests/test_utils/test_timer.py

21 lines
362 B
Python
Raw Normal View History

import time
from mmdeploy.utils.timer import TimeCounter
def test_count_time():
class test:
@TimeCounter.count_time()
def fun1(self):
time.sleep(0.01)
t = test()
with TimeCounter.activate('fun1', warmup=10, log_interval=10):
for i in range(50):
t.fun1()
for i in range(50):
t.fun1()