@Bhack
Thank you for the link, helpful. As I also mentioned in the query that we can accomplish on_epoch_start/end, on_batch_start/end timing.
Actually, I was interested in timing on  model_forward, model_backward ,  AFAIK,  under normal circumstances, we can’t use callback for that, can we? (However, I will test other options using the callbacks to verify the timing results).