ChatRWKV
ChatRWKV copied to clipboard
include "timeit" module in benchmark.py
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.
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
cbe00741b0732cfe9c7b27924811903bd566cb98