A-bone1 / Attention-ocr-Chinese-Version

Attention OCR Based On Tensorflow
431 stars 141 forks source link

您好,我修改了这个代码,吧cnn的size该小了,但是没办法用tf1.10跑程序,似乎无论我吧size怎么改小,都会报OOM #55

Open tnkong opened 5 years ago

tnkong commented 5 years ago

程序是正常运行的,能看见loss 运行几百个step后 然后就OOM了,说是一个矩阵乘法需要更多的内存,但是内存耗尽了。无论我吧cnn层提取的feature map修改的多么小 这个都会出