How to save / load part of checkpoint?

❓ Questions/Help/Support

Hi @vfdev-5 ,

I have a feature request from Clara users for transfer learning: I trained a model with 4 output classes, then I want to load the checkpoint for a 3 classes task. How to save weights except for the last output layer in ignite Checkpoint handler? Or skip the last layer when loading checkpoint in the handler?

Thanks in advance.

[Nic-Ma]

Hi @vfdev-5 ,

Thanks for your detailed solutions! I prefer the first solution because we may not know the exact layers affected by class numbers, for example, maybe a different input channel number will affect several convolution layers. Maybe we can enhance the MONAI CheckpointLoader with the strick_shape option later, I submitted ticket: https://github.com/Project-MONAI/MONAI/issues/1939

Thanks.

[vfdev-5]

I think we can close this issue as solved.