-
Notifications
You must be signed in to change notification settings - Fork 7.6k
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
表格识别 训练时候,进行eval 时候遇到bug #13268
Comments
运行环境是什么 paddle版本是多少 paddleocr 版本是多少 |
paddleocr 是2.8 ,paddlepaddle-gpu == 0.0.0.post112 |
用的是啥gpu呀,amp模式训练的吗,不用amp可以运行吗 |
4090,就是正常模式。把那个数据格式改成统一的int32就可以执行。。不是这个bug吗? |
好的 @liuzhipengchd ,谢谢反馈,你能提一个PR来修复它吗? |
因为这里看上去是启用amp,导致的bug。 |
好的。。大佬,我想问下。1、训练表格识别模型,那个效果好 SLANet_lcnetv2.yml 和 SLANet_ch.yml (如果我选择scale 2.5的呢),2、SLANet_lcnetv2 中 我想选择 PPLCNetV2_large的话,应该怎么配置 |
应该是 |
@liuzhipengchd 无法复现你的问题 python3 tools/train.py -c configs/table/SLANet.yml -o Train.loader.batch_size_per_card=16 Eval.loader.batch_size_per_card=16
python3 tools/train.py -c configs/table/SLANet.yml -o Train.loader.batch_size_per_card=16 Eval.loader.batch_size_per_card=16 Global.use_amp=True
|
我这里可以复现问题 |
@liuzhipengchd 不会影响效果,这个已经修复了。 |
PaddleOCR/ppocr/modeling/heads/table_att_head.py
Line 370 in 7a3c580
[Hint: Expected data_type == phi::DataType::FLOAT16 || data_type == phi::DataType::BFLOAT16 == true, but received data_type == phi::DataType::FLOAT16 || data_type == phi::DataType::BFLOAT16:0 != true:1.] (at ../paddle/fluid/imperative/amp_auto_cast.cc:190)
在评估时候,structure_ids 的数据类型和pre_chars 不一致,导致bug
The text was updated successfully, but these errors were encountered: