search

AndriyMulyar / bert_document_classification

architectures and pre-trained models for long document classification.
154 stars 47 forks source link

Using the Roberta model training by transformers api (checkpoint) #17

Open davendw49 opened 3 years ago

davendw49 commented 3 years ago

I want to change the bert model, and I changed the config.ini in order to use my own model. but my model was training by transformers api and got the file list like(actually it's a checkpoint) as following the screenshot QQ拼音截图20201127210058 After using your code or after my try to change your code into transformers version, It got the same error:

so, I wonder whether it's caused by the model is only a checkpoint? image

hope to have your reply~ thank you!