Hi @nivesh_gadipudi, here are a couple examples for using Ray Tune with Pytorch 2.
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!