Hugging Face's logo

LLM 课程文档

当您遇到错误时该怎么办

Hugging Face's logo
加入 Hugging Face 社区

并获得增强的文档体验

开始

当您遇到错误时该怎么办

Ask a Question Open In Colab Open In Studio Lab

在本节中，我们将了解在尝试从您新调整的 Transformer 模型生成预测时可能发生的一些常见错误。这将为您准备第 4 节，在那里我们将探讨如何调试训练阶段本身。

我们为本节准备了一个模板模型仓库，如果您想运行本章中的代码，您首先需要将模型复制到您在 Hugging Face Hub 上的帐户。为此，请首先通过在 Jupyter 笔记本中运行以下代码登录

from huggingface_hub import notebook_login

notebook_login()

或者在您喜欢的终端中运行以下代码

huggingface-cli login

这将提示您输入用户名和密码，并将令牌保存在 ~/.cache/huggingface/ 下。登录后，您可以使用以下函数复制模板仓库

from distutils.dir_util import copy_tree
from huggingface_hub import Repository, snapshot_download, create_repo, get_full_repo_name


def copy_repository_template():
    # Clone the repo and extract the local path
    template_repo_id = "lewtun/distilbert-base-uncased-finetuned-squad-d5716d28"
    commit_hash = "be3eaffc28669d7932492681cd5f3e8905e358b4"
    template_repo_dir = snapshot_download(template_repo_id, revision=commit_hash)
    # Create an empty repo on the Hub
    model_name = template_repo_id.split("/")[1]
    create_repo(model_name, exist_ok=True)
    # Clone the empty repo
    new_repo_id = get_full_repo_name(model_name)
    new_repo_dir = model_name
    repo = Repository(local_dir=new_repo_dir, clone_from=new_repo_id)
    # Copy files
    copy_tree(template_repo_dir, new_repo_dir)
    # Push to Hub
    repo.push_to_hub()

现在，当您调用 copy_repository_template() 时，它将在您的帐户下创建模板仓库的副本。

调试来自 🤗 Transformers 的流程

为了开始我们进入调试 Transformer 模型的美妙世界之旅，请考虑以下情景：您正在与一位同事合作一个问答项目，以帮助电子商务网站的客户找到有关消费类产品的答案。您的同事向您发送了一条消息，例如

大家好！我刚刚使用 Hugging Face 课程第 7 章中的技术进行了一个实验，并在 SQuAD 上获得了一些很棒的结果！我认为我们可以使用这个模型作为我们项目的起点。Hub 上的模型 ID 是 “lewtun/distillbert-base-uncased-finetuned-squad-d5716d28”。请随意测试一下 :)

您首先想到的就是使用 🤗 Transformers 的 pipeline 加载模型

from transformers import pipeline

model_checkpoint = get_full_repo_name("distillbert-base-uncased-finetuned-squad-d5716d28")
reader = pipeline("question-answering", model=model_checkpoint)
"""
OSError: Can't load config for 'lewtun/distillbert-base-uncased-finetuned-squad-d5716d28'. Make sure that:

- 'lewtun/distillbert-base-uncased-finetuned-squad-d5716d28' is a correct model identifier listed on 'https://huggingface.co/models'

- or 'lewtun/distillbert-base-uncased-finetuned-squad-d5716d28' is the correct path to a directory containing a config.json file
"""

哦不，似乎出错了！如果您是编程新手，这些类型的错误起初可能看起来有点神秘（OSError 到底是什么？！）。此处显示的错误只是一个更大的错误报告（称为 Python 回溯（也称为堆栈跟踪））的最后一部分。例如，如果您在 Google Colab 上运行此代码，您应该会看到如下截图

A Python traceback.

这些报告中包含大量信息，因此让我们一起了解关键部分。首先要注意的是，回溯应从下往上阅读。如果您习惯于从上到下阅读英文文本，这听起来可能很奇怪，但这反映了这样一个事实：回溯显示了 pipeline 在下载模型和分词器时进行的一系列函数调用。（有关 pipeline 在底层如何工作的更多详细信息，请查看第 2 章。）

🚨 看到 Google Colab 回溯中 “6 frames” 周围的蓝色框了吗？这是 Colab 的一项特殊功能，它将回溯压缩为 “frames”。如果您似乎找不到错误的来源，请务必单击那两个小箭头展开完整的回溯。

这意味着回溯的最后一行指示了最后的错误消息，并给出了引发的异常的名称。在本例中，异常类型为 OSError，表示与系统相关的错误。如果我们阅读随附的错误消息，我们可以看到模型的 config.json 文件似乎存在问题，并且我们得到了两个修复建议

"""
Make sure that:

- 'lewtun/distillbert-base-uncased-finetuned-squad-d5716d28' is a correct model identifier listed on 'https://huggingface.co/models'

- or 'lewtun/distillbert-base-uncased-finetuned-squad-d5716d28' is the correct path to a directory containing a config.json file
"""

💡 如果您遇到难以理解的错误消息，只需将消息复制并粘贴到 Google 或 Stack Overflow 搜索栏中（是的，真的！）。您很有可能不是第一个遇到此错误的人，这是一种查找社区中其他人发布的解决方案的好方法。例如，在 Stack Overflow 上搜索 OSError: Can't load config for 会得到几个 点击，可以作为解决问题的起点。

第一个建议是要求我们检查模型 ID 是否确实正确，因此首要任务是复制标识符并将其粘贴到 Hub 的搜索栏中

The wrong model name.

嗯，看起来我们同事的模型确实不在 Hub 上……啊哈，但是模型名称中有一个错字！DistilBERT 的名称中只有一个 “l”，所以让我们修复它，并搜索 “lewtun/distilbert-base-uncased-finetuned-squad-d5716d28”

The right model name.

好的，这次找到了。现在让我们尝试使用正确的模型 ID 再次下载模型

model_checkpoint = get_full_repo_name("distilbert-base-uncased-finetuned-squad-d5716d28")
reader = pipeline("question-answering", model=model_checkpoint)
"""
OSError: Can't load config for 'lewtun/distilbert-base-uncased-finetuned-squad-d5716d28'. Make sure that:

- 'lewtun/distilbert-base-uncased-finetuned-squad-d5716d28' is a correct model identifier listed on 'https://huggingface.co/models'

- or 'lewtun/distilbert-base-uncased-finetuned-squad-d5716d28' is the correct path to a directory containing a config.json file
"""

唉，又失败了——欢迎来到机器学习工程师的日常生活！既然我们已经修复了模型 ID，那么问题肯定出在仓库本身。访问 🤗 Hub 上仓库内容的快速方法是通过 huggingface_hub 库的 list_repo_files() 函数

from huggingface_hub import list_repo_files

list_repo_files(repo_id=model_checkpoint)
['.gitattributes', 'README.md', 'pytorch_model.bin', 'special_tokens_map.json', 'tokenizer_config.json', 'training_args.bin', 'vocab.txt']

有趣——仓库中似乎没有 config.json 文件！难怪我们的 pipeline 无法加载模型；我们的同事一定是忘了在微调后将此文件推送到 Hub。在这种情况下，问题似乎很容易解决：我们可以要求他们添加文件，或者，由于我们可以从模型 ID 中看到使用的预训练模型是 distilbert-base-uncased，我们可以下载此模型的 config 并将其推送到我们的仓库，看看是否可以解决问题。让我们试试看。使用我们在第 2 章中学到的技术，我们可以使用 AutoConfig 类下载模型的配置

from transformers import AutoConfig

pretrained_checkpoint = "distilbert-base-uncased"
config = AutoConfig.from_pretrained(pretrained_checkpoint)

🚨 我们在这里采取的方法并非万无一失，因为我们的同事可能在微调模型之前调整了 distilbert-base-uncased 的配置。在现实生活中，我们应该先与他们确认，但为了本节的目的，我们将假设他们使用了默认配置。

然后我们可以使用配置的 push_to_hub() 函数将其推送到我们的模型仓库

config.push_to_hub(model_checkpoint, commit_message="Add config.json")

现在我们可以通过从 main 分支的最新提交加载模型来测试这是否有效

reader = pipeline("question-answering", model=model_checkpoint, revision="main")

context = r"""
Extractive Question Answering is the task of extracting an answer from a text
given a question. An example of a question answering dataset is the SQuAD
dataset, which is entirely based on that task. If you would like to fine-tune a
model on a SQuAD task, you may leverage the
examples/pytorch/question-answering/run_squad.py script.

🤗 Transformers is interoperable with the PyTorch, TensorFlow, and JAX
frameworks, so you can use your favourite tools for a wide variety of tasks!
"""

question = "What is extractive question answering?"
reader(question=question, context=context)
{'score': 0.38669535517692566,
 'start': 34,
 'end': 95,
 'answer': 'the task of extracting an answer from a text given a question'}

哇呼，成功了！让我们回顾一下您刚刚学到的内容

  • Python 中的错误消息称为回溯，应从下往上阅读。错误消息的最后一行通常包含您需要定位问题来源的信息。
  • 如果最后一行不包含足够的信息，请向上查看回溯，看看是否可以确定错误发生在源代码的哪个位置。
  • 如果所有错误消息都无法帮助您调试问题，请尝试在线搜索类似问题的解决方案。
  • huggingface_hub // 🤗 Hub? 库提供了一套工具，您可以使用这些工具与 Hub 上的仓库进行交互和调试。

现在您已经知道如何调试 pipeline，让我们来看看模型本身前向传播中一个更棘手的示例。

调试模型的前向传播

虽然 pipeline 非常适合大多数需要快速生成预测的应用，但有时您需要访问模型的 logits（例如，如果您想应用一些自定义的后处理）。为了了解在这种情况下可能出现什么问题，让我们首先从我们的 pipeline 中获取模型和分词器

tokenizer = reader.tokenizer
model = reader.model

接下来我们需要一个问题，所以让我们看看是否支持我们最喜欢的框架

question = "Which frameworks can I use?"

正如我们在第 7 章中看到的，我们需要采取的常用步骤是标记化输入、提取开始和结束标记的 logits，然后解码答案跨度

import torch

inputs = tokenizer(question, context, add_special_tokens=True)
input_ids = inputs["input_ids"][0]
outputs = model(**inputs)
answer_start_scores = outputs.start_logits
answer_end_scores = outputs.end_logits
# Get the most likely beginning of answer with the argmax of the score
answer_start = torch.argmax(answer_start_scores)
# Get the most likely end of answer with the argmax of the score
answer_end = torch.argmax(answer_end_scores) + 1
answer = tokenizer.convert_tokens_to_string(
    tokenizer.convert_ids_to_tokens(input_ids[answer_start:answer_end])
)
print(f"Question: {question}")
print(f"Answer: {answer}")
"""
---------------------------------------------------------------------------
AttributeError                            Traceback (most recent call last)
/var/folders/28/k4cy5q7s2hs92xq7_h89_vgm0000gn/T/ipykernel_75743/2725838073.py in <module>
      1 inputs = tokenizer(question, text, add_special_tokens=True)
      2 input_ids = inputs["input_ids"]
----> 3 outputs = model(**inputs)
      4 answer_start_scores = outputs.start_logits
      5 answer_end_scores = outputs.end_logits

~/miniconda3/envs/huggingface/lib/python3.8/site-packages/torch/nn/modules/module.py in _call_impl(self, *input, **kwargs)
   1049         if not (self._backward_hooks or self._forward_hooks or self._forward_pre_hooks or _global_backward_hooks
   1050                 or _global_forward_hooks or _global_forward_pre_hooks):
-> 1051             return forward_call(*input, **kwargs)
   1052         # Do not call functions when jit is used
   1053         full_backward_hooks, non_full_backward_hooks = [], []

~/miniconda3/envs/huggingface/lib/python3.8/site-packages/transformers/models/distilbert/modeling_distilbert.py in forward(self, input_ids, attention_mask, head_mask, inputs_embeds, start_positions, end_positions, output_attentions, output_hidden_states, return_dict)
    723         return_dict = return_dict if return_dict is not None else self.config.use_return_dict
    724
--> 725         distilbert_output = self.distilbert(
    726             input_ids=input_ids,
    727             attention_mask=attention_mask,

~/miniconda3/envs/huggingface/lib/python3.8/site-packages/torch/nn/modules/module.py in _call_impl(self, *input, **kwargs)
   1049         if not (self._backward_hooks or self._forward_hooks or self._forward_pre_hooks or _global_backward_hooks
   1050                 or _global_forward_hooks or _global_forward_pre_hooks):
-> 1051             return forward_call(*input, **kwargs)
   1052         # Do not call functions when jit is used
   1053         full_backward_hooks, non_full_backward_hooks = [], []

~/miniconda3/envs/huggingface/lib/python3.8/site-packages/transformers/models/distilbert/modeling_distilbert.py in forward(self, input_ids, attention_mask, head_mask, inputs_embeds, output_attentions, output_hidden_states, return_dict)
    471             raise ValueError("You cannot specify both input_ids and inputs_embeds at the same time")
    472         elif input_ids is not None:
--> 473             input_shape = input_ids.size()
    474         elif inputs_embeds is not None:
    475             input_shape = inputs_embeds.size()[:-1]

AttributeError: 'list' object has no attribute 'size'
"""

哦，糟糕，看起来我们的代码中有一个错误！但我们不怕一点调试。您可以在笔记本中使用 Python 调试器

或在终端中

在这里，读取错误消息告诉我们 'list' object has no attribute 'size'，我们可以看到一个 --> 箭头指向 model(**inputs) 中引发问题的行。您可以使用 Python 调试器以交互方式调试此问题，但现在我们将简单地打印出 inputs 的切片，看看我们有什么

inputs["input_ids"][:5]
[101, 2029, 7705, 2015, 2064]

这看起来确实像一个普通的 Python list，但让我们仔细检查类型

type(inputs["input_ids"])
list

是的，确实是 Python list。那么哪里出错了？回想一下第 2 章，🤗 Transformers 中的 AutoModelForXxx 类在张量（PyTorch 或 TensorFlow 中的任何一个）上运行，常见的操作是使用 Tensor.size()（例如在 PyTorch 中）提取张量的维度。让我们再看一下回溯，看看哪一行触发了异常

~/miniconda3/envs/huggingface/lib/python3.8/site-packages/transformers/models/distilbert/modeling_distilbert.py in forward(self, input_ids, attention_mask, head_mask, inputs_embeds, output_attentions, output_hidden_states, return_dict)
    471             raise ValueError("You cannot specify both input_ids and inputs_embeds at the same time")
    472         elif input_ids is not None:
--> 473             input_shape = input_ids.size()
    474         elif inputs_embeds is not None:
    475             input_shape = inputs_embeds.size()[:-1]

AttributeError: 'list' object has no attribute 'size'

看起来我们的代码试图调用 input_ids.size()，但这显然不适用于 Python list，它只是一个容器。我们如何解决这个问题？在 Stack Overflow 上搜索错误消息会得到很多相关的点击。单击第一个会显示一个与我们类似的问题，答案如下面的屏幕截图所示

An answer from Stack Overflow.

答案建议我们在分词器中添加 return_tensors='pt'，所以让我们看看这是否对我们有效

inputs = tokenizer(question, context, add_special_tokens=True, return_tensors="pt")
input_ids = inputs["input_ids"][0]
outputs = model(**inputs)
answer_start_scores = outputs.start_logits
answer_end_scores = outputs.end_logits
# Get the most likely beginning of answer with the argmax of the score
answer_start = torch.argmax(answer_start_scores)
# Get the most likely end of answer with the argmax of the score
answer_end = torch.argmax(answer_end_scores) + 1
answer = tokenizer.convert_tokens_to_string(
    tokenizer.convert_ids_to_tokens(input_ids[answer_start:answer_end])
)
print(f"Question: {question}")
print(f"Answer: {answer}")
"""
Question: Which frameworks can I use?
Answer: pytorch, tensorflow, and jax
"""

太棒了，成功了！这是一个很好的例子，说明 Stack Overflow 是多么有用：通过识别类似的问题，我们能够从社区中其他人的经验中受益。但是，这样的搜索并不总是会产生相关的答案，那么在这种情况下您该怎么办呢？幸运的是，在 Hugging Face 论坛上有一个热情的开发者社区可以帮助您！在下一节中，我们将了解如何编写好的论坛问题，这些问题很可能得到解答。

< > 在 GitHub 上更新

←介绍 在论坛上寻求帮助→