The post_mp weights couldn't get removed from the pretrained ckpt when the reset_prediction_head parameter was set True because all keys started with 'model'. I fixed the issue by first removing the post_mp layer weights and then adding 'model' to key names in pretrained_dict.
The post_mp weights couldn't get removed from the pretrained ckpt when the reset_prediction_head parameter was set True because all keys started with 'model'. I fixed the issue by first removing the post_mp layer weights and then adding 'model' to key names in pretrained_dict.