Make profiling possible on validation steps.
🚀 Make profiling possible on validation steps.
It would be really nice, if we could also profile function calls at validation and/or test stage.
Motivation
In my case the implementation and functionality of the validation step is really different than the one from the training step so it would be nice to also profile this one. Different metrics are getting calculated and some other stuff is going on.