如何改成一次生成多条输出语句呢？

Morizeyao / GPT2-Chinese

Chinese version of GPT2 training code, using BERT tokenizer.

MIT License

7.48k stars 1.7k forks source link

Open viego1999 opened 2 years ago

viego1999 commented 2 years ago

generate.py中batch_size循环生成的out都是同一个值，应该是让返回的outs包含batch_size条生成的结果，请问如何更改？