search

PaddlePaddle / PaddleNLP

👑 Easy-to-use and powerful NLP and LLM library with 🤗 Awesome model zoo, supporting wide-range of NLP tasks from research to industrial applications, including 🗂Text Classification, 🔍 Neural Search, ❓ Question Answering, ℹ️ Information Extraction, 📄 Document Intelligence, 💌 Sentiment Analysis etc.
https://paddlenlp.readthedocs.io
Apache License 2.0
12.04k stars 2.93k forks source link

text_classification\pretrained_models\train.py训练出现异常 #772

Closed LangDaoAI closed 3 years ago

LangDaoAI commented 3 years ago

请Team看一下，当使用train.py训练时候出现如下异常， 使用的数据集是chnsenticorp

image

LangDaoAI commented 3 years ago

我是修改了源码，变更为ernie-gram-zh预训练模型

image

LangDaoAI commented 3 years ago

image

image

LangDaoAI commented 3 years ago

我设置了batch是16，max length是256

linjieccc commented 3 years ago

您好，可以换成ppnlp.transformers.ErnieGramForSequenceClassification再试下

LangDaoAI commented 3 years ago

image

换了之后，出现上面错误

LangDaoAI commented 3 years ago

我是从以下地址下载了新的ernie_gram_zh模型，放到了$HOME/paddlenlp/models/ernie_gram_zh目录下面，

https://github.com/PaddlePaddle/PaddleNLP/blob/develop/paddlenlp/transformers/ernie_gram/modeling.py#L106

LangDaoAI commented 3 years ago

已经解决了，关闭问题