-
Notifications
You must be signed in to change notification settings - Fork 1.5k
Loading models with strict=False #1891
Copy link
Copy link
Closed
Labels
enhancementNew feature or requestNew feature or request
Description
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.
Reactions are currently unavailable
Metadata
Metadata
Assignees
Labels
enhancementNew feature or requestNew feature or request