Expose load_state_dict strict=False
🚀 Feature
In contrastive learning, we normally train a representation learning backbone then adding the classifier. Sometimes, I wish to play with different classifiers. It would be best to have strict=False exposed to make the model still load with a user's permission.
model = TransferLearningModel.load_from_checkpoint(ckpt_path, strict=False)
The implementation is intuitive but an extremely useful feature in my case. Thank you!