search

RUCAIBox / TextBox

TextBox 2.0 is a text generation library with pre-trained language models
https://github.com/RUCAIBox/TextBox
MIT License
1.07k stars 117 forks source link

Text Generation Using SeqGAN #300

Closed hmqt closed 1 year ago

hmqt commented 1 year ago

Hi, After I trained the SeqGAN Model on Arabic texts I have a generated file (SeqGAN-wiki_ar-Dec-05-2022_20-05-03.txt) which contains a huge number of Arabic-generated sentences. But my question to you is, is it possible for me, after the end of the training period for this model (which was approximately three days), to run it in order to get a new sentence that was just generated every time, immediately? Does TextBox tools support this feature and how can I run the SeqGAN after training on the special dataset to generate a new sentence every time?

StevenTang1998 commented 1 year ago

If you has already trained a model in the saved folder named SeqGAN-dataset-time.pth, you can load it and just generate sentences by adding the command --test_only=True --load_experiment=saved/SeqGAN-dataset-time.pt to your original command.

hmqt commented 1 year ago

Thank you so much