Chinese version of GPT2 training code, using BERT tokenizer.
-
Updated
Mar 17, 2022 - Python
Add a description, image, and links to the text-generation topic page so that developers can more easily learn about it.
To associate your repository with the text-generation topic, visit your repo's landing page and select "manage topics."
setting
pretrained_model_name
will not only define the model arch but also load the pre-trained checkpoint. We should have anotherhparam
to control whether to load pre-trained checkpoint or not.