Skip to content

Lightning 2.5.5 breaks checkpoint reuseΒ #291

@ziw-liu

Description

@ziw-liu

We have workflows where checkpoints trained with one wrapper class is reused for another. For example, a FullyConvolutionalMAE wrapped in FcmaeUNet during training could be wrapped in AugmentedPredictionVSUNet for inference. Lightning-AI/pytorch-lightning#21116 shipped in lightning 2.5.5 breaks this use case as it raises an exception if the hyperparameter does not match what's saved in the checkpoint.

Metadata

Metadata

Assignees

No one assigned

    Labels

    bugSomething isn't workingtranslationImage translation (VS)

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions