gpt4 book ai didi

sentence-similarity - 如何在训练后在本地保存 SetFit 训练器

转载 作者:行者123 更新时间:2023-12-02 01:27:46 26 4
gpt4 key购买 nike

我在 HPC 上工作,工作节点上没有互联网访问权限,训练后保存 SetFit 训练器的唯一选择是将其推送到 HuggingFace hub。如何将其保存到本地磁盘?

https://github.com/huggingface/setfit

最佳答案

setfit有这个类方法

model._save_pretrained(save_directory)

并加载它

saved_model = SetFitModel._from_pretrained(save_directory)

关于sentence-similarity - 如何在训练后在本地保存 SetFit 训练器,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/74046426/

26 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com