

TITLE
Adding call back to measure time of each step 

COMMENTS
1

REACTIONS
+1: 0
-1: 0
laugh: 0
hooray: 0
heart: 0
rocket: 0
eyes: 0

BODY
Hi
I need to measure the time of each training step, with distributed version of seq2seq_trainer with TPUs, could you provide me with some hints how to measure the time? thanks 