pytorch - dropout()：参数“输入”（位置 1）必须是张量，而不是当使用带有 Huggingface 的 Bert 时的 str

Question

我的代码运行良好，当我今天尝试在不更改任何内容的情况下运行它时，出现以下错误：

dropout(): argument 'input' (position 1) must be Tensor, not str

如果能提供帮助将不胜感激。
可能是数据加载器的问题？

标签： pytorchbert-language-model

Answer 1

如果您使用 HuggingFace，此信息可能很有用。我有同样的错误，并通过在退出之前在模型类中添加参数return_dict=False来修复它：输出 = 模型（**输入，return_dict=False）