Early Stop in Grid Search

[zixiqin]

Hi there,

I'm a beginner with KAN, and I'm trying to use grid search to find the best parameters. Everything is going well using a simple for loop to handle the grid search. However, I'm stuck on choosing the right value for "Step" for each grid search experiment. I'm trying to find a way to implement an early stopping mechanism, where the model will capture the best step when its performance does not improve for a certain number of steps (for example, at least 5 steps without improvement).

Does anyone know how to apply early stopping, similar to what most neural network models can do, in KAN?

Thank you so much!