search

YCG09 / chinese_ocr

CTPN + DenseNet + CTC based end-to-end Chinese OCR implemented using tensorflow and keras
Apache License 2.0
2.74k stars 1.08k forks source link

keras 训练densenet模型速度慢 #190

Open wqt2019 opened 5 years ago

wqt2019 commented 5 years ago

环境tf-gpu1.11，keras2.2.0，之前训练densenet的时候速度还可以，今天重新训练的时候速度很慢，使用的是作者提供的300万的数据，不知道问题出在哪了 Epoch 1/10 686/28184 [..............................] - ETA: 10:21:46 - loss: 29.8879 - acc: 0.0270

ericxbji commented 5 years ago

环境tf-gpu1.11，keras2.2.0，之前训练densenet的时候速度还可以，今天重新训练的时候速度很慢，使用的是作者提供的300万的数据，不知道问题出在哪了 Epoch 1/10 686/28184 [..............................] - ETA: 10:21:46 - loss: 29.8879 - acc: 0.0270

想问下，你用作者提供的数据重新训练，是在作者给的weights_densenet.h5基础上开始重新训练的，还是从头开始训练到，我现在拿作者提供的数据训练一直不收敛

@evanfly

wqt2019 commented 5 years ago

重新训练的，除了３００万的数据还有自己生成的一些，可以收敛的，你看下标签有没有问题，或者改小一点学习率。 自问自答一下，正常训练变慢了不知道原因，然后model.fit_generator加了多进程训练，速度稍微快了一点

mayidu commented 5 years ago

您好,请问怎么解决这个问题呢？ @ericxbji