因为Optuna 目前分布式训练只能在.db文件下进行,所以需要将现在已经保存过的.pkl文件转化为.db文件,经过在Github上询问,得到如下思路:
- 首先,训练一个study,保存为.pkl文件:
import optuna
import joblib
def objective(trial):
x = trial.suggest_float("x", -1, 1)
return x ** 2
study = optuna.create_study()
study.optimize(objective, n_trials=3)
joblib