如何在训练后将 SetFit 训练器保存到本地

Tan*_*fna 3 sentence-similarity sentence-transformers huggingface

我正在开发一台工作节点上无法访问互联网的 HPC,训练后保存 SetFit 训练器的唯一选择是将其推送到 HuggingFace 中心。我如何将其保存到本地磁盘?

https://github.com/huggingface/setfit

use*_*139 9

setfit 有这个类方法

model._save_pretrained(save_directory)
Run Code Online (Sandbox Code Playgroud)

并加载它

saved_model = SetFitModel._from_pretrained(save_directory)
Run Code Online (Sandbox Code Playgroud)