ChatRWKV icon indicating copy to clipboard operation
ChatRWKV copied to clipboard

include "timeit" module in benchmark.py

Open bennmann opened this issue 2 years ago • 2 comments
trafficstars

I'm not exactly sure the best way to implement this, however it would be good to use "timeit" python module in the todo loop at the end of benchmark.py somehow to show how long each 100 model.forward loop takes.

bennmann avatar Mar 03 '23 02:03 bennmann

why not just use time.time()?

import time
start_time = time.time()
#do something you need to time
end_time = time.time()
print(end_time-start_time) #time from start to end

Durham avatar Mar 03 '23 15:03 Durham

cbe00741b0732cfe9c7b27924811903bd566cb98

bennmann avatar Mar 10 '23 00:03 bennmann