使用optuna保存pytorch中的最佳模型



有没有一种简单的方法可以从pytorch中的optuna研究中保存最佳模型?

如果使用LightGBMTuner,则可以使用LightGBMTuner.get_best_booster方法来获得最佳模型。否则,您需要自己保存经过训练的模型(c.f.,如何保存在目标函数中训练的机器学习模型?(。

参考:https://github.com/optuna/optuna/issues/2085

最新更新