Hi senmao! I’m writting you because I tried to get your results by my own… but, I cound’t get them. I have the same as you posted and I’m getting the time by this code:
for i in range(1000):
t1 = time.time()
out, hid = native_lstm(inp)
t2 = time.time()
t_native += t2-t1
t3 = time.time()
out2, hid2 = custom_lstm(inp, [(hx, cx)])
t4 = time.time()
t_custom += t4-t3
Thanks for all!
Regards,
David.