Skip to content

Loading models with strict=False #1891

@danieltudosiu

Description

@danieltudosiu

Is your feature request related to a problem? Please describe.
In some cases we might want to load partial models. this can be done by loading a state_dict with argument strict=False as per: https://pytorch.org/tutorials/beginner/saving_loading_models.html#warmstarting-model-using-parameters-from-a-different-model

Describe the solution you'd like
Allow users to pass kwargs to ignite's Checkpoint handlers as per: https://pytorch.org/ignite/handlers.html#ignite.handlers.Checkpoint.load_objects

Describe alternatives you've considered
None, as MONAI does not allow alternatives besides using Ignite directly.

Metadata

Metadata

Assignees

Labels

enhancementNew feature or request

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions