使用中英文训练模型报内存溢出,使用ch_PP-OCRv3_rec.yml配置文件开启训练

[sssjc666]

请提供下述完整信息以便快速定位问题/Please provide the following information to quickly locate the problem 该模型之前在4g显存上正常运行过，现在换了一个电脑想试一下8g运存发现无法启动训练。模型选用ocr官网上公布pp-ocrv3中英文训练模型，数据为自己的数据，batch_size设置为1时，也无法启动训练。希望大佬回复！

• 系统环境/System Environment：win11
• 版本号/Version：Paddle：2.5.1 PaddleOCR：2.7.0.2 问题相关组件/Related components：paddleocr
• 运行指令/Command Code：python tools/train.py -c configs/rec/PP-OCRv3/ch_PP-OCRv3_rec_distillation.yml -o Global.pretrained_model=./pretrained_model/ch_PP-OCRv3_rec_train/best_accuracy
• 完整报错/Complete Error Message： MemoryError:

---

C++ Traceback (most recent call last):

Not support stack backtrace yet.

---

Error Message Summary:

ResourceExhaustedError:

Out of memory error on GPU 0. Cannot allocate 128.000000MB memory on GPU 0, 7.995605GB memory has been allocated and available memory is only 0.000000B.

Please check whether there is any other process using GPU 0.

1. If yes, please stop them, or start PaddlePaddle on another GPU.
2. If no, please decrease the batch size of your model. (at ..\paddle\fluid\memory\allocation\cuda_allocator.cc:86)

我们提供了AceIssueSolver来帮助你解答问题，你是否想要它来解答(请填写yes/no)?/We provide AceIssueSolver to solve issues, do you want it? (Please write yes/no):

请尽量不要包含图片在问题中/Please try to not include the image in the issue.

[sssjc666]

配置详情如下： 希望大佬回复！

[sssjc666]

已经解决

[sssjc666]

当时可能是环境出现问题了，对ocr库进行了一次重装，再次进行训练就能正常启动。ps：使用paddle不建议win11进行练习，我遇见好几次包的问题了

------------------ 原始邮件 ------------------ 发件人: "PaddlePaddle/PaddleOCR" @.>; 发送时间: 2023年11月20日(星期一) 中午11:24 @.>; @.@.>; 主题: Re: [PaddlePaddle/PaddleOCR] 使用中英文训练模型报内存溢出,使用ch_PP-OCRv3_rec.yml配置文件开启训练 (Issue #11266)

请问怎么解决的呢

— Reply to this email directly, view it on GitHub, or unsubscribe. You are receiving this because you authored the thread.Message ID: @.***>