Cannot finetune en_ppstructure_mobile_v2.0_SLANet_train with paddlepaddle-GPU

5jdjgkvh  于 5个月前  发布在  其他
关注(0)|答案(1)|浏览(44)
问题描述 / Problem Description

Finetuning en_ppstructure_mobile_v2.0_SLANet_train with paddlepaddle-GPU fails with error.
Finetuning with the CPU package works just fine.

运行环境 / Runtime Environment
  • OS: Windows 10, Windows 11
  • Paddle: 2.6.0, 2.6.1
  • PaddleOCR: 2.7.5
  • CUDA/CUDNN: 11.7/8.9.7, 12.0/8.9.7
复现代码 / Reproduction Code

Dataset and annotation structure: PubTabNet
Loss function: SLALoss

完整报错 / Complete Error Message

(InvalidArgument) Currently, the type of tensor in slice indices only allows int32 and int64, please check the type of index tensor. (at ..\paddle/fluid/pybind/slice_utils.h:70)

可能解决方案 / Possible solutions

Finetuning with the CPU package works just fine.

附件 / Appendix
mrwjdhj3

mrwjdhj31#

The GPU works for SLANet now after downgrading to PaddleOCR 2.6.0
Problem persists for 2.7.5

相关问题