自己写的动态图 CRNN 模型训练一直不正确
Created by: xiangyubo
1)PaddlePaddle版本:1.6.2 3)GPU:GPU v100 4)系统环境:Python3.7,AI Studio 在线 notebook 环境
- 训练信息 1)单机单卡 2)显存信息,16G显存,32G内存
我自己照着静态图的 CRNN 和一些动态图的模型,自己写了一个动态图版本的 CRNN 模型。流程能跑通。训练时候 loss 看着很正常,第一个 batch 的 loss 是40+,运行20个 batch 之后下降到7-8,之后继续下降到4-5。但是解码出来的东西一直不对,字段级别的准确率一直是0。 解码部分我一开始用的是 fluid.layers.ctc_greedy_decoder,结果不对。我觉得这个太复杂,自己写了一个类似的,但依然不对。求帮忙看看是哪个部分不对,导致准确率一直为0。
AI Studio 环境地址:https://aistudio.baidu.com/aistudio/projectdetail/251819 里面带着数据集和完整代码,用GPU方式运行,进去解压数据直接训练就可以。项目已经公开