PaddleNLP [Bug]: 'NoneType' 对象没有 'from_pretrained' 属性 - 超长文本分类任务训练报错

slhcrj9b  于 2个月前  发布在  其他
关注(0)|答案(1)|浏览(34)

软件环境

- paddlepaddle:
- paddlepaddle-gpu: 
- paddlenlp:

重复问题

  • 我已搜索现有的问题

错误描述

[2024-06-18 17:39:15,317] [    INFO] - tokenizer config file saved in /root/.paddlenlp/models/ernie-doc-base-zh/tokenizer_config.json
[2024-06-18 17:39:15,317] [    INFO] - Special tokens file saved in /root/.paddlenlp/models/ernie-doc-base-zh/special_tokens_map.json
Traceback (most recent call last):
  File "/data/llms/github_code/PaddleNLP-release-2.8/examples/text_classification/ernie_doc/train.py", line 345, in <module>
    do_train(args)
  File "/data/llms/github_code/PaddleNLP-release-2.8/examples/text_classification/ernie_doc/train.py", line 181, in do_train
    model = ErnieDocForSequenceClassification.from_pretrained(args.model_name_or_path, num_classes=num_classes)
  File "/data/llms/github_code/PaddleNLP-release-2.8/paddlenlp/transformers/model_utils.py", line 2087, in from_pretrained
    config, model_kwargs = cls.config_class.from_pretrained(
AttributeError: 'NoneType' object has no attribute 'from_pretrained'

稳定复现步骤 & 代码

python train.py --batch_size 4 
--model_name_or_path ernie-doc-base-zh 
--epoch 5 
--output_dir ./checkpoints/
2ul0zpep

2ul0zpep1#

我这边可以正常加载,不会有你的这个报错。你的paddlenlp版本是什么呢?看看能不能更新一下。

相关问题