AttributeError: ‘list‘ object has no attribute ‘size‘问题解决

Tina_1024

已于 2022-10-24 11:39:55 修改

阅读量5.1k

点赞数 1

CC 4.0 BY-SA版权

分类专栏： NLP python 文章标签： python 开发语言 1024程序员节

于 2022-10-23 21:30:01 首次发布

本文链接：https://blog.csdn.net/sinat_36023271/article/details/127481271

python 同时被 2 个专栏收录

7 篇文章

订阅专栏

NLP

3 篇文章

订阅专栏

博客讲述了在使用预训练模型时遇到的AttributeError，原因是将list传递给了需要tensor的模型。通过在tokenizer中添加`return_tensors=pt`参数，成功将list转换为PyTorch tensor，从而解决了问题。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

刚看到这个错误提示的时候感到很奇怪，为什么会出现这个报错信息，我的代码如下：

import sys
sys.path.append("../input/iterativestratification")
import pandas as pd
import numpy as np
from transformers import AutoTokenizer,AutoModel

tokenizer = AutoTokenizer.from_pretrained(DEBERTA_BASE)
model = AutoModel.from_pretrained(DEBERTA_BASE)

tokens = tokenizer(essay)
input_ids = tokens['input_ids']
model_output = model(input_ids)

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

仔细思考了以下，应是input_ids是list类型，衡量list类型的大小可以用len()函数，但是在model_DB中，需要用到tensor作为输入，所以在tokenizer时，添加return_tensors="pt"，代码如下：

tokens = tokenizer(essay,return_tensors="pt")
input_ids = tokens['input_ids']
model_output = model(input_ids)

问题解决。