我在 HPC 上工作,工作节点上没有互联网访问权限,训练后保存 SetFit 训练器的唯一选择是将其推送到 HuggingFace hub。如何将其保存到本地磁盘?
最佳答案
setfit有这个类方法
model._save_pretrained(save_directory)
并加载它
saved_model = SetFitModel._from_pretrained(save_directory)
关于sentence-similarity - 如何在训练后在本地保存 SetFit 训练器,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/74046426/