Matching ray-tune validation with training validation

@kai I just posted another question relevant to this, with a self-contained minimal example