使用UIE训练doccano数据集时,提示无法创建张量
收藏
paddleNLP版本:PaddleNLP-2.4.5
运行脚本:
python finetune.py --train_path ./data/train.txt --dev_path ./data/train.txt --output_dir ./checkpoint/model_best --model uie-base --learning_rate 1e-5 --batch_size 16 --max_seq_len 5602 --num_train_epochs 10 --seed 1000 --logging_steps 10 --device "gpu" --do_train --do_eval --overwrite_output_dir
报错:
TypeError: forward() got an unexpected keyword argument 'pos_ids'
ValueError: Unable to create tensor, you should probably activate truncation and/or padding with 'padding=True' 'truncation=True' to have batched tensors with the same length
0
收藏
请登录后评论