Utilities for partially loading weights from a state_dict

**Is your feature request related to a problem? Please describe.**
A utility function is needed to set a model's state_dict partially from another one.
this is required by https://github.com/Project-MONAI/MONAI/issues/2273 and https://github.com/Project-MONAI/MONAI/discussions/2329

an implementation is already part of the checkpoint loader:
https://github.com/Project-MONAI/MONAI/blob/a5364621712fe6b3de584223a4e28be638bc0fc4/monai/handlers/checkpoint_loader.py#L95-L106


	k, _ = list(self.load_dict.items())[0]
	# single object and checkpoint is directly a state_dict
	if len(self.load_dict) == 1 and k not in checkpoint:
	checkpoint = {k: checkpoint}

	# skip items that don't match data shape
	for k, obj in self.load_dict.items():
	if isinstance(obj, (nn.DataParallel, nn.parallel.DistributedDataParallel)):
	obj = obj.module
	if isinstance(obj, torch.nn.Module):
	d = obj.state_dict()
	checkpoint[k] = {k: v for k, v in checkpoint[k].items() if k in d and v.shape == d[k].shape}

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

Utilities for partially loading weights from a state_dict #2339

Metadata

Assignees

Labels

Type

Projects

Milestone

Relationships

Development

Utilities for partially loading weights from a state_dict #2339

Description

Metadata

Metadata

Assignees

Labels

Type

Projects

Milestone

Relationships

Development

Issue actions