konumaru / commonLit_readability_prize

https://www.kaggle.com/c/commonlitreadabilityprize
0 stars 0 forks source link

tokenizerの出力を `**inputs` で入力できるように変更する #9

Closed konumaru closed 3 years ago

konumaru commented 3 years ago

下記のようにするために、CommonLitDataset の return をinputsにまとめる

>>> inputs = tokenizer("Hello, my dog is cute", return_tensors="pt")
>>> outputs = model(**inputs)