This is a Pytorch implementation of NewsGC
python train.py
The script has been tested running under Python 3.6.13/DeepIn 20.2.4, with the following packages installed (along with their dependencies):
- pytorch==1.7.1
- tensorflow_gpu==2.4.0
- transformers==4.12.5
- sentencepiece==0.1.95
- rouge-metric==1.0.1