抱脸变压器在哪里保存模型？

Question

运行以下代码将下载模型-有人知道它将其下载到哪个文件夹吗？

!pip install -q transformers
from transformers import pipeline
model = pipeline('fill-mask')

Answer 1

This post应该有所了解（加上我自己的研究，因为它已经比较老了）。

如前所述，Linux系统中的默认位置为 ~/.cache/torch/transformers/ （目前，我使用的是Translators 2.7，但不太可能很快更改）。该目录中的神秘文件夹名称似乎与Amazon S3哈希相对应。

还请注意，管道任务只是“路由”到其他模型。要了解您当前正在加载哪一个，请参见here。对于您的特定模型，pipeline(fill-mask)实际上使用了distillroberta-base模型。

Answer 2

从 Transformers 4.3 版开始，缓存位置已更改。

确切位置在此代码部分中定义 https://github.com/huggingface/transformers/blob/master/src/transformers/file_utils.py#L181-L187

在 Linux 上，它位于 ~/.cache/huggingface/transformers。

那里的文件名基本上是下载文件的原始 URL 的 SHA 哈希值。对应的json文件可以帮你搞清楚原来的文件名是什么

Answer 3

在 Windows 10 上，将 ~ 替换为 C:\Users\username 或在 cmd 中执行 cd /d "%HOMEDRIVE%%HOMEPATH%"。

所以完整路径将是：C:\Users\username\.cache\huggingface\transformers