huggingface预训练模型保存格式
时间: 2023-10-13 11:07:50 浏览: 356
Hugging Face实战(NLP实战/Transformer实战/预训练模型/模型微调/PyTorch版本)下篇之模型训练
huggingface预训练模型的保存格式是PyTorch的`.pt`文件或TensorFlow的`.h5`文件。可以使用`torch.save()`函数保存PyTorch模型为`.pt`文件,使用`model.save_pretrained()`方法保存Transformers模型为TensorFlow的`.h5`文件。
阅读全文