Support PyTorch Lightning 2.3

cellarium-ai / cellarium-ml

Distributed single-cell data analysis.

BSD 3-Clause "New" or "Revised" License

11 stars 2 forks source link

Closed ordabayevy closed 1 month ago

ordabayevy commented 2 months ago

Fixes:

Returning None from training step is not allowed during distributed training https://github.com/Lightning-AI/pytorch-lightning/pull/19918
Breaking change introduced by the support for dependency injection (we instead rely on initialization under meta device context) https://github.com/Lightning-AI/pytorch-lightning/pull/18105