PaddlePaddle / X2Paddle

Deep learning model converter for PaddlePaddle. (『飞桨』深度学习模型转换工具)
http://www.paddlepaddle.org/
Apache License 2.0
727 stars 164 forks source link

转pytorch到paddle报错 #1008

Open 18575062745 opened 6 months ago

18575062745 commented 6 months ago

问题描述

具体信息

代码:
from x2paddle.convert import pytorch2paddle
pytorch2paddle(module=model,
                        save_dir="/home/zhaowenzhuo/project/2024/baidu_race_true/beam_retriever-main/output/best_checkpoint/checkpoint_best.pt",
                        jit_type="trace",
                        input_examples=(batch['input_ids'],batch['attention_mask'],batch['start_pos'],
                                        batch['end_pos'],batch['sentence_index'],batch['sentence_labels'],batch['answer_type']))
Jiang-Jia-Jun commented 6 months ago

存在自定义Python算子,是无法支持的