search

optuna / optuna-examples

Examples for https://github.com/optuna/optuna
MIT License
632 stars 171 forks source link

Support Lighting instead/besides Pytorch Lighting #220

Closed afcarzero1 closed 8 months ago

afcarzero1 commented 8 months ago

Provide and integration module for the new Lighting framework instead of PyTorch Lighting

Motivation

The new framework is lighting. Using the old callback optuna.integration.PyTorchLightningPruningCallback does not work with the new framework (I guess it mixes imports , which leads to bugs as explained here

If your feature request is related to a problem, please describe a clear and concise description of what the problem is. -->

Exceptions when using the new Lighting package.

The new callback should extend the behavior of the current one but for the new Lighting package

afcarzero1 commented 8 months ago

I see it is actually fixed. For some reason I do not obtain it in my optuna installation when doing pip install optuna.