Using trained model with Skorch

[btvd]

Community, is there a way to configure skorch wrapper for PyTorch trained model (no need to use fit method) to add only compatibility with sklearn ecosystem?

[BenjaminBossan]

Yes, that's absolutely possible. For instance, check this notebook. It defines a BertModule that wraps a Hugging Face transformers model. The wrapped model can be used with skorch as usual, whether it be for fine-tuning or just for inference.

[btvd]

Thank you very much for hint!