Releases: jacksonchen1998/Empowering-NLG
Releases · jacksonchen1998/Empowering-NLG
Final version
What we add
Add paper url and source code.
Program information
dataprepare.py
: Data preprocessinginfer_gpt.py
: Infer GPT-2 model with unput textinfer_score.py
: Infer score with input textmodel.py
: Score model based on GPT-2 and RoBERTatrainRL.py
: Train PPO model, using GPT-2 and score modeltraingpt.py
: Train GPT-2 modeltrainscore.py
: Train score model