Lightning v2.5.4

Changes in 2.5.4

PyTorch Lightning

Fixed

• Fixed AsyncCheckpointIO snapshots tensors to avoid race with parameter mutation (#21079)
• Fixed AsyncCheckpointIO threadpool exception if calling fit or validate more than one (#20952)
• Fixed learning rate not being correctly set after using LearningRateFinder callback (#21068)
• Fixed misalignment column while using rich model summary in DeepSpeedstrategy (#21100)
• Fixed RichProgressBar crashing when sanity checking using val dataloader with 0 len (#21108)

Lightning Fabric

Changed

• Added support for NVIDIA H200 GPUs in get_available_flops (#20913)

Full commit list: 2.5.3 -> 2.5.4

Contributors

We thank all folks who submitted issues, features, fixes and doc changes. It's the only way we can collectively make Lightning ⚡ better for everyone, nice job!

In particular, we would like to thank the authors of the pull-requests above, in no particular order:

@fnhirwa, @GdoongMathew, @jjh42, @littlebullGit, @SkafteNicki

Thank you ❤️ and we hope you'll keep them coming!