search

VDIGPKU / IterNet

14 stars 1 forks source link

Error When Training Model with Custom Dataset #2

Closed hoho303 closed 2 years ago

hoho303 commented 2 years ago

Hello. Thank you for your great work. I try to training ITerNet in VinText dataset with Vietnamese characters such as ("Ạ", "Â", "Ơ",...). Firstly, I train the language model using my own vietnamese token but got nan loss. Can you help me to fix this ?

image

chuxiaojie commented 2 years ago

Hi Our IterNet is based on language model proposed by ABINet. Some useful information about this issue can be found in FangShancheng/ABINet#13.

Besides, you can also try to train IterNet without loading pretrained language model.

hoho303 commented 2 years ago

Thank for your response. I am trying to train your model without loading pretrained language model. So I will close this issue.