加載擁抱臉轉換器時出錯 model 以執行文本分類

Question

我正在按照本筆記本中所述的定期訓練來訓練變壓器 model，以使用預期答案 class 對問題進行分類。

訓練完model后，想看一些題的預測，於是寫了如下代碼：

from transformers import pipeline,AutoModel, AutoModelForSequenceClassification, AutoTokenizer

model_name = 'aubmindlab/bert-base-arabertv02'

arabert_model = AutoModelForSequenceClassification.from_pretrained('/gdrive/MyDrive/LabelModel')

tokenizer = AutoTokenizer.from_pretrained(model_name)

text = "أين وقعت غزوة بدر؟"
#{'كيان': 0, 'تقريري': 1, 'حدث': 2, 'رقم': 3, 'عاقل': 4, 'موقع': 5, 'وصف': 6}

pipe = pipeline(model=arabert_model, tokenizer=tokenizer)

pipe(text) 

首先，它給了我這個錯誤：

OSError: Error no file named pytorch_model.bin, tf_model.h5, model.ckpt.index or flax_model.msgpack found in directory /gdrive/MyDrive/LabelModel.

然后，我按照此鏈接中的建議，通過將目錄中的 training_args.bin 文件重命名為 pytorch_model.bin 來解決了這個問題。

之后，存在另一個錯誤：

AttributeError: 'TrainingArguments' object has no attribute 'keys'

我該如何解決這個問題？

Answer 1

問題出在Google Drive空間，它刪除了pytorch.bin文件。 所以bin文件相關的代碼沒有錯誤。