`Step` timings include `testing` timings

[Tialo]

Hi! The step timing includes testing time, making it appear too large in plots. In my case testing during training takes significantly longer, the scale for step is not ideal. If it's okay with you, I can open PR that makes step independent of testing.

Example

[Tialo]

proposed change should also fix some other metrics