首页
学习
活动
专区
圈层
工具
发布
首页
学习
活动
专区
圈层
工具
MCP广场
社区首页 >问答首页 >如何用拥抱的脸来微调一个模型?

如何用拥抱的脸来微调一个模型?
EN

Stack Overflow用户
提问于 2022-09-03 22:10:02
回答 1查看 107关注 0票数 -1

我想下载一个预先训练过的模型,并用我自己的数据对模型进行微调。我已经下载了一个伯特-大纳模型文物从拥抱脸,我已经列出了以下内容。对于这一点,我想知道我需要哪些文件或工件,从外观上看,pytorch_model.bin是经过训练的模型,但这些其他文件是什么以及它们的用途,比如令牌程序文件和vocab.txt .

代码语言:javascript
运行
复制
config.json
pytorch_model.bin
special_tokens_map.json
tokenizer_config.json
vocab.txt
EN

回答 1

Stack Overflow用户

发布于 2022-09-07 02:45:44

这些不同的文件是模型的元数据和正在使用的令牌程序(当您序列化模型时,这是输出)。要从HF集线器微调一个预先训练过的模型,您可以使用PyTorch或TF,也可以使用训练员类,在那里您不必编写自己的自定义培训代码。例如:

代码语言:javascript
运行
复制
trainer = Trainer(
    model=model,
    args=training_args,
    train_dataset=small_train_dataset,
    eval_dataset=small_eval_dataset,
    compute_metrics=compute_metrics,
)

参考这里的官方文档,了解如何对经过预先培训的模型进行端到端的调优:https://huggingface.co/docs/transformers/training

票数 1
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/73595680

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档