Skip to content

首先感谢分享,请问预训练模型方便告知一下吗?或者说能分享下从零开始训练的方法。 另外能适配下mps吗 #46

Answered by OleehyO
z20150214 asked this question in Q&A
Discussion options

You must be logged in to vote

TexTeller没有使用预训练模型,模型架构类似于TrOCR,也可以看一下TexTeller.py

训练方法就写在train.py里,其实就是端到端训练,没有什么特别的。

mps适配的话之后会考虑加进去。

Replies: 2 comments

Comment options

You must be logged in to vote
0 replies
Answer selected by OleehyO
Comment options

You must be logged in to vote
0 replies
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Category
Q&A
Labels
None yet
2 participants
Converted from issue

This discussion was converted from issue #46 on June 07, 2024 06:19.