Skip to content

Fine tuning PP-OCRv5_mobile_rec runs out of memory on 5080 #17486

@ncihnegn

Description

@ncihnegn

🔎 Search before asking

  • I have searched the PaddleOCR Docs and found no similar bug report.
  • I have searched the PaddleOCR Issues and found no similar bug report.
  • I have searched the PaddleOCR Discussions and found no similar bug report.

🐛 Bug (问题描述)

But it works for PP-OCRv5_server_rec.

🏃‍♂️ Environment (运行环境)

OS:  Arch Linux kernel 6.18.4
Python: 3.13.11
PaddleOCR: 3.3.1

🌰 Minimal Reproducible Example (最小可复现问题的Demo)

python tools/train.py -c configs/rec/PP-OCRv5/PP-OCRv5_mobile_rec.yml -o Global.pretrained_model=./PP-OCRv5_mobile_rec_pretrained.pdparams

Metadata

Metadata

Assignees

Labels

perf/oomOut of Memory (OOM) issue during model training or inference.

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions