Skip to content

Commit

Permalink
add note for fixing pruning callback issue (#1575)
Browse files Browse the repository at this point in the history
Co-authored-by: madtoinou <32447896+madtoinou@users.noreply.github.com>
  • Loading branch information
dennisbader and madtoinou committed Feb 26, 2023
1 parent f0ea7e5 commit d6a742b
Showing 1 changed file with 2 additions and 0 deletions.
2 changes: 2 additions & 0 deletions docs/userguide/hyperparameter_optimization.md
Original file line number Diff line number Diff line change
Expand Up @@ -12,6 +12,8 @@ For the sake of the example, we train a `TCNModel` on a single series, and optim
You can also have a look at [this notebook](https://github.com/unit8co/darts/blob/master/examples/17-hyperparameter-optimization.ipynb)
for a more complete example.

_Note (2023-19-02): Optuna's `PyTorchLightningPruningCallback` raises an error with pytorch-lightning>=1.8. Until this fixed, a workaround is proposed [here](https://github.com/optuna/optuna-examples/issues/166#issuecomment-1403112861)._

```python
import numpy as np
import optuna
Expand Down

0 comments on commit d6a742b

Please sign in to comment.