我正在AzureML的笔记本上运行HuggingFace微调。当我在本地运行下面的代码时，我得到了保存在相应目录中的各种文件（模型bin文件，一堆json）。
trainer.train() trainer.save_model()
但是，当我在AzureML vm上执行相同操作时，尽管我在计算文件夹结构中看到了目录，但它是空的。如何保存模型？
z2acfund1#
正如@Ro.oT所提到的，您可以通过指定输出目录来使用trainer.save_model()。
trainer.save_model()
或者您可以使用trainer.push_to_hub()在您的集线器中共享它。请参阅此在运行上面的代码之前，请确保使用下面的代码登录到您的中心。
trainer.push_to_hub()
from huggingface_hub import notebook_login notebook_login()
1条答案
按热度按时间z2acfund1#
正如@Ro.oT所提到的，您可以通过指定输出目录来使用
trainer.save_model()。或者您可以使用
trainer.push_to_hub()在您的集线器中共享它。请参阅此在运行上面的代码之前，请确保使用下面的代码登录到您的中心。