Revive test-time augmentation

[MillionIntegrals]

Because of a rework of metrics system I had to disable test-time augmentation code from the framework.

It should be put back somewhere, maybe in a different spot as the previous one proved to be very hard to implement in a generic way.