Uncertainty: steps: Added a wait for computation on validation set