Lightning-AI/pytorch-lightning 2.2.2

Patch release v2.2.2

on GitHub

PyTorch

Fixed

• Fixed an issue causing a TypeError when using torch.compile as a decorator (#19627)
• Fixed a KeyError when saving a FSDP sharded checkpoint and setting save_weights_only=True (#19524)

Fabric

Fixed

• Fixed an issue causing a TypeError when using torch.compile as a decorator (#19627)
• Fixed issue where some model methods couldn't be monkeypatched after being Fabric wrapped (#19705)
• Fixed an issue causing weights to be reset in Fabric.setup() when using FSDP (#19755)

Full Changelog: 2.2.1...2.2.2

Contributors

@ankitgola005 @awaelchli @Borda @carmocca @dmitsf @dvoytan-spark @fnhirwa