Pytorch Lightning Integration

Open wisecornelius opened this issue 3 years ago • 3 comments

This is a feature request:

It would be great to have an integration with PyTorch Lightning the way that Optuna, Ray-Tune or W&B have it.

Jun 14 '22 10:06 wisecornelius

Thanks for the suggestion. Does the "way Ray-Tune have it" refer to this?

Jun 14 '22 14:06 sonichi

@sonichi yes, as far as I remember, all that is necessary is the lightning callback to report training metrics for pruning.

Jun 14 '22 16:06 wisecornelius

@sonichi yes, as far as I remember, all that is necessary is the lightning callback to report training metrics for pruning.

Good. Since flaml.tune can use ray as the backend, you can do the same thing in flaml.tune.run() as long as setting use_ray=True.

Jun 14 '22 17:06 sonichi