问题描述 / 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.
1条答案
按热度按时间mrwjdhj31#
The GPU works for SLANet now after downgrading to PaddleOCR 2.6.0
Problem persists for 2.7.5