运行如下代码的时候出现标题上的报错,将wandb更新到最新版本或者降为老版本都无法解决。
from transformers import Trainer
trainer = Trainer(
model,
training_args,
train_dataset=tokenized_datasets["train"],
eval_dataset=tokenized_datasets["validation"],
data_collator=data_collator,
tokenizer=tokenizer,
)
trainer.train()
原因是protobuf版本太高
pip install protobuf==3.20.0
将版本更新为3.20.0即可解决