RUCAIBox/TextBox

Ran Model after training

Closed this issue · 4 comments

hmqt commented

how i can run my mode after training it on my data

Could you provide more detailed information?

hmqt commented

i trained my dataset (novels) on SeqGAN model and i got this file SeqGAN-novels-Apr-23-2023_23-30-28.pth. How i can used this file to generate arabic texts any time any moment.

You can add --test_only=True --load_experiment=saved/SeqGAN-novels-Apr-23-2023_23-30-28.pth to your initial command. It will generate as many examples as your test set.

我同样想知道怎么在训练之后运行模型,但我训练完之后并没有得到.pth文件。我的训练命令是 python run_textbox.py --model=BART --dataset=samsum --model_path=facebook/bart-base。得到一个checkpoint_best文件夹,请教一下怎么使用训练得到的模型