问题描述
使用darts做时间序列预测时,使用optuna做超参数优化,调用study.optimize(objective, timeout=7200),报错:ValueError: Expected a parent
原因
新版本的pytorch_lightning改名为lightning,在函数objective中采用了pytorch_lightning的EarlyStopping的回调函数,同时采用了from optuna.integration import PyTorchLightningPruningCallback的PyTorchLightningPruningCallback回调函数,在optuna的3.5版本中采用了lightning.pytorch来代替pytorch_lightning,因此同时存在pytorch_lightning和lightning.pytorch两个新老版本引发了冲突。
解决方式
- 定义OptunaPruning类替代PyTorchLightningPruningCallback,使两个版本兼容(不推荐)。
class OptunaPruning(PyTorchLightningPruningCallback, pytorch_lightning.Callback)
Optuna与pytorch_lightning版本冲突的解决策略,

最低0.47元/天 解锁文章
3177

被折叠的 条评论
为什么被折叠?



