Early stopping with Optim when the validation loss increases

I’m using Optim.jl to train a model but it keeps overfiting. is there nice way of implementing a early stopping callback that has the current x? I tried setting the callback in Optim.Options but i only get the iteration, loss and time. is it possible to access other optimizer states?

Yes, it should work if you set extended_trace=true in the options

1 Like