小编Joh*_*pac的帖子

使用 HuggingFace 微调 T5 进行汇总时出现关键错误

我正在尝试微调 T5 变压器以进行汇总,但收到一条关键错误消息:

KeyError: 'Indexing with integers (to access backend Encoding for a given batch index) is not available when using Python based tokenizers'
Run Code Online (Sandbox Code Playgroud)

我使用的代码基本上是这样的:

model_name = '...'
tokenizer = T5Tokenizer.from_pretrained(model_name)
model = T5ForConditionalGeneration.from_pretrained(model_name)
model.to(device)

(...)

df_dataset = df_dataset[['summary','document']]
df_dataset.document = 'summarize: ' + df_dataset.document

X = list(df_dataset['document'])
y = list(df_dataset['summary'])
X_train, X_val, y_train, y_val = train_test_split(X, y, test_size=0.2)
X_train_tokenized = tokenizer(X_train, padding=True, truncation=True, max_length=512)
y_train_tokenized = tokenizer(y_train, padding=True, truncation=True, max_length=512)
X_val_tokenized = tokenizer(X_val, padding=True, truncation=True, max_length=512)
y_val_tokenized = tokenizer(y_val, …
Run Code Online (Sandbox Code Playgroud)

python huggingface-transformers huggingface-tokenizers

7
推荐指数
1
解决办法
7619
查看次数