g2pW icon indicating copy to clipboard operation
g2pW copied to clipboard

模型训练问题——简体/繁体

Open Xwmiss opened this issue 2 years ago • 9 comments

感谢您的工作! 我最近想要重新训练一下g2pW模型,但是我在看issue和cpp_dataset中的数据时有一个问题,在issue中好像讨论的都是当下的模型是基于繁体中文来训练的;而cpp_dataset中的数据却都是简体中文;那么整体训练过程到底是使用了简体还是繁体呢?

期待您的回复。

Xwmiss avatar Sep 26 '23 07:09 Xwmiss

顺便问一下,我下载了readme中的文件,如下G2PWModel-v2.zip。但是该如何使用呢,在scripts/predict_g2p_bert.py的使用过程中,如下: image image 以上问题请问该如何解决。

Xwmiss avatar Sep 26 '23 10:09 Xwmiss

目前可下載的模型是基於繁體中文資料所訓練而成,其資料品質較好,而資料不開源。

而資料的格式採用和g2pM相同的格式,因此你可以自行使用g2pM訓練一個簡中模型,訓練方法如README所示。

GitYCC avatar Sep 26 '23 13:09 GitYCC

please refer: https://github.com/GitYCC/g2pW#load-offline-model

GitYCC avatar Sep 26 '23 13:09 GitYCC

please refer: https://github.com/GitYCC/g2pW#load-offline-model

感谢您的回复,我已明白了训练中的简体/繁体问题。但是,#load-offline-model中并没能解答我的困惑,G2PWConverter中传入的是onnx模型,而我想使用您所开源的pth模型,但是根据#predict,我却没办法直接使用您训练得到的pth权重

Xwmiss avatar Sep 27 '23 00:09 Xwmiss

please refer: https://github.com/GitYCC/g2pW#load-offline-model

感谢您的回复,我已明白了训练中的简体/繁体问题。但是,#load-offline-model中并没能解答我的困惑,G2PWConverter中传入的是onnx模型,而我想使用您所开源的pth模型,但是根据#predict,我却没办法直接使用您训练得到的pth权重

我想我有点明白了,您的意思是我可以不直接使用pth权重? 我看到了您开源代码中的g2pW/scripts/export_onnx_model.py,看来在训练以后,不选择pth而是直接将pth权重转为onnx进行调用?

Xwmiss avatar Sep 27 '23 00:09 Xwmiss

目前可下載的模型是基於繁體中文資料所訓練而成,其資料品質較好,而資料不開源。

而資料的格式採用和g2pM相同的格式,因此你可以自行使用g2pM訓練一個簡中模型,訓練方法如README所示。

我想再打扰您一下,训练所得到的模型是简中/繁中的设置是完全依赖于训练数据吗?也就是说如果我给的是简体的训练数据,出来的模型就是简中模型,若是繁中数据,得到的结果就是繁中模型?

此外,如果我想要的同样也是繁中模型,按照这个逻辑,我应该可以用opencc把简体中文的数据全部处理一遍来得到繁中数据,从而训练得到我所需要的繁中模型?

Xwmiss avatar Sep 27 '23 01:09 Xwmiss

你可以從pretrained model進行微調

GitYCC avatar Sep 27 '23 05:09 GitYCC

你可以從pretrained model進行微調

我是想尝试bert的结构做的小一些,所以无法使用pretrained model进行微调。

Xwmiss avatar Sep 27 '23 05:09 Xwmiss

你可以從pretrained model進行微調

我是想尝试bert的结构做的小一些,所以无法使用pretrained model进行微调。

目前看起来,如果我想要使用cpp数据集来训练得到一个类似的繁中模型,我需要使用opencc对cpp模型的文本进行处理,随后处理pinyin到bopomofo的转换,这样应该就可以训练得到对应的模型了。

Xwmiss avatar Sep 27 '23 06:09 Xwmiss