用最好的optuna试用保存模型

Question

我使用 optuna (PyTorch) 训练了一个模型，其中试验次数为 5。

这就是我的输出：

现在，第四次试验是最好的试验。我不只是想要最佳参数值，我想用这些值保存模型以供 lr 和优化器使用。我想知道如何做到这一点。

Answer 1

在 Optuna 的官方常见问题解答中，您可以找到以下内容：

https://optuna.readthedocs.io/en/stable/faq.html#how-to-save-machine-learning-models-trained-in-objective-functions