Open 18575062745 opened 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']))
存在自定义Python算子,是无法支持的
问题描述
具体信息