ch_PP-OCRv4_rec.yml 这个配置我怎么训练不了,batch size = 1都爆显存。 ch_PP-OCRv4_rec.yml 这个配置没问题,可以训练 Collaborator GreatV commented Jul 6, 2024 感觉得检查一下 ch_PP-OCRv4_rec_svtr_large.yml 里的参数跟 tools/infer/predict_rec.py 里的是不是一致的。 GreatV added need ...
paddlepaddle-gpu 2.5.2 paddleslim 从release/2.5安装的 PaddleOCR git checkout release/2.7 git checkout5219872 修改的代码: 因为报错, 所以修改了export_model.py和quant.py, 不确定修改的对不对 git diff deploy/slim/quantization/export_model.py diff --git a/deploy/slim/quantization/export_model.py b...
RecResizeImg不加确实不通过,要么会报batch因为图片size不同无法拼接,我看文档v4使用了不同尺度的输入训练,这里我怀疑是没适配好。sampler里面也新增了一个多尺度的。 最后这个是svtr teacher模型似乎出了问题,正常来说svtr的ctc head是不需要额外的svtr模块的,lcnet的版本才需要,我猜测是要把ctc head的svtr部分改...
configs/det/ch_PP-OCRv4/ch_PP-OCRv4_rec_distill.yml 这份配置文件,我修改了数据集的路径,指向我本地电脑上的路据集。 Contributor jiqirenfeile commented Aug 19, 2024 It appears that the following issue is related to the one I'm currently facing, so I will continue troubleshooting from this...