batch_size, seq_length = input_shape ValueError: not enough values to unpack (expected 2, got 1)

Ang_Quantum

于 2021-11-23 10:02:43 发布

阅读量4.1k

点赞数 10

分类专栏：日常bug类文章标签： pytorch python

本文链接：https://blog.youkuaiyun.com/Ang_Quantum/article/details/121486890

版权

这个bug发生于写BERT的在线inference代码。错误如下：

Traceback (most recent call last):
  File "classifier.py", line 94, in <module>
    result = clf.predict(content)
  File "classifier.py", line 77, in predict
    outputs = self.model(ids, mask, token_type_ids)
  File "/data/angchen/env/torch/lib/python3.6/site-packages/torch/nn/modules/module.py", line 493, in __call__
    result = self.forward(*input, **kwargs)
  File "../src/bert.py", line 29, in forward
    _, output_1 = self.l1(ids, attention_mask = mask, token_type_ids=token_type_ids)
  File "/data/angchen/env/torch/lib/python3.6/site-packages/torch/nn/modules/module.py", line 493, in __call__
    result = self.forward(*input, **kwargs)
  File "/data/angchen/env/torch/lib/python3.6/site-packages/transformers/models/bert/modeling_bert.py", line 950, in forward
    batch_size, seq_length = input_shape
ValueError: not enough values to unpac