Closed rishi-a closed 1 year ago
Resolved it temporarily by upgrading Pytorch Lightning to the latest version. Not sure if that was the actual solution to the problem as well.
I ran into the same issue. Updated Pytorch-lightning. However, I can not use optuna with the latest version of PyTorch-lightning.
Yes, the PyTorch-Lightning (PTL) version was most likely the root cause for this issue. I believe PTL versions shortly before 2.0.0 should work fine as well.
@helenehanyu it should work if you replace Optuna'sPyTorchLightningPruningCallback
with the one proposed in our example description here (the link is not formatted well in docs unfortunately, I'll fix that soon. Here is the link).
Thank you @dennisbader , now it works.
Describe the bug Fitting the NBeats model results in
RuntimeError: No precision set
To Reproduce
model = NBEATSModel(input_chunk_length=24, output_chunk_length=12, random_state=42)
model.fit([train_air_scaled, train_milk_scaled], epochs=50, verbose=True);