I have trained a model using optuna (PyTorch) where number of trials was 5.
This is what my output looks like :
Now, the 4th trial is the best trial. I dont just want the best parameter values, I want to save the model with these values for lr and optimizer. I want to know how that can be done.