2024-09-16|閱讀時間 ‧ 約 23 分鐘

AI說書 - 從0開始 - 178 | 保存訓練好的模型

我想要一天分享一點「LLM從底層堆疊的技術」,並且每篇文章長度控制在三分鐘以內,讓大家不會壓力太大,但是又能夠每天成長一點。


整理目前手上有的素材:


既然模型訓練好了,自然需要執行保存的動作:

trainer.save_model("./KantaiBERT")


結果為:


當中和模型相關的檔案是:

  • config.json
  • pytorch_model.bin
  • training_args.bin


而和 Pretrained Tokenization 相關的檔案是 (詳見 AI說書 - 從0開始 - 166 | 保存訓練好的 Tokenizer):

  • merges.txt
  • vocab.json


分享至
成為作者繼續創作的動力吧!
© 2024 vocus All rights reserved.