How to calculate inference time

[Thangtran27]

I read your paper about inference time. But I don't find the code to carry out calculating inference time. So how can I do that?
I'm looking forward to hearing from you!

[nagadit]

Hello @Thangtran27 !!
You can measure the inference time using this code!

time_list = []
for _ in range(100):
 st = time.time()
 out = model(input)
 time_list.append(time.time() - st)

print(np.mean(time_list)