Fix validation callback

It seems the validation callback in hysteresis-scripts does not work well for multi-gpu training. This fixes Investigate reason for predicting wrong.

• Fix validation callback for multi-gpu training [priority:: low] [completion:: 2024-04-17]