Hey @nivesh_gadipudi, just wanted to repost here (I work on Ray Tune):
An easy way to prevent overfitting is simply to return “done=True” when validation losses begins to diverge.
tune.report(done=True, ...)

# or if using class API:
def step(self):
   ...
   return {"done": True, ...}

Hope that helps!