Tan*_*fna 3 sentence-similarity sentence-transformers huggingface
我正在开发一台工作节点上无法访问互联网的 HPC,训练后保存 SetFit 训练器的唯一选择是将其推送到 HuggingFace 中心。我如何将其保存到本地磁盘?
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)
| 归档时间: |
|
| 查看次数: |
2876 次 |
| 最近记录: |