im2latex项目环境搭建
原项目地址
原始版本地址https://github.com/guillaumegenthial/im2latex
项目环境搭建
由于原始项目github写的比较抽象,并且有些坑,特此记录一下这个项目的搭建过程。
首先python版本是2.7,tensorflow版本是1.4.1,如果使用gpu训练,cuda版本是8.0,使用配套的cudnn即可。nltk版本3.4.5,其他库可以使用最新版。
训练过程
原始超参数batchsize较大,我使用的是2080ti,显存12g,batchsize>4时在训练中会遇到OOM错误,调低batchsize即可。
也可以直接用torch版,环境搭建比较简单,torch版地址(准确度较低)https://github.com/luopeixiang/im2latex