├── .gitignore ├── LICENSE ├── README.md ├── image ├── HKUST.jpg └── pytorch-logo-dark.png ├── requirements.txt ├── scripts ├── dapt_pretraining.sh ├── finetuning.sh ├── sdpt_pretraining.sh └── tapt_pretraining.sh └── src ├── cal_rouge.py ├── dapt_pretraining.py ├── inference.py ├── others ├── __init__.py ├── logging.py ├── my_pyrouge.py ├── optimizer.py ├── recadam.py └── utils.py ├── preprocessing.py ├── run.py ├── sdpt_pretraining.py ├── tapt_pretraining.py └── trainer.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TysonYu/AdaptSum/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TysonYu/AdaptSum/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TysonYu/AdaptSum/HEAD/README.md -------------------------------------------------------------------------------- /image/HKUST.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TysonYu/AdaptSum/HEAD/image/HKUST.jpg -------------------------------------------------------------------------------- /image/pytorch-logo-dark.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TysonYu/AdaptSum/HEAD/image/pytorch-logo-dark.png -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | numpy 2 | tqdm 3 | transformers>=3.0.2 4 | nltk -------------------------------------------------------------------------------- /scripts/dapt_pretraining.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TysonYu/AdaptSum/HEAD/scripts/dapt_pretraining.sh -------------------------------------------------------------------------------- /scripts/finetuning.sh: -------------------------------------------------------------------------------- 1 | python ./src/run.py -------------------------------------------------------------------------------- /scripts/sdpt_pretraining.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TysonYu/AdaptSum/HEAD/scripts/sdpt_pretraining.sh -------------------------------------------------------------------------------- /scripts/tapt_pretraining.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TysonYu/AdaptSum/HEAD/scripts/tapt_pretraining.sh -------------------------------------------------------------------------------- /src/cal_rouge.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TysonYu/AdaptSum/HEAD/src/cal_rouge.py -------------------------------------------------------------------------------- /src/dapt_pretraining.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TysonYu/AdaptSum/HEAD/src/dapt_pretraining.py -------------------------------------------------------------------------------- /src/inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TysonYu/AdaptSum/HEAD/src/inference.py -------------------------------------------------------------------------------- /src/others/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/others/logging.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TysonYu/AdaptSum/HEAD/src/others/logging.py -------------------------------------------------------------------------------- /src/others/my_pyrouge.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TysonYu/AdaptSum/HEAD/src/others/my_pyrouge.py -------------------------------------------------------------------------------- /src/others/optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TysonYu/AdaptSum/HEAD/src/others/optimizer.py -------------------------------------------------------------------------------- /src/others/recadam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TysonYu/AdaptSum/HEAD/src/others/recadam.py -------------------------------------------------------------------------------- /src/others/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TysonYu/AdaptSum/HEAD/src/others/utils.py -------------------------------------------------------------------------------- /src/preprocessing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TysonYu/AdaptSum/HEAD/src/preprocessing.py -------------------------------------------------------------------------------- /src/run.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TysonYu/AdaptSum/HEAD/src/run.py -------------------------------------------------------------------------------- /src/sdpt_pretraining.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TysonYu/AdaptSum/HEAD/src/sdpt_pretraining.py -------------------------------------------------------------------------------- /src/tapt_pretraining.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TysonYu/AdaptSum/HEAD/src/tapt_pretraining.py -------------------------------------------------------------------------------- /src/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TysonYu/AdaptSum/HEAD/src/trainer.py --------------------------------------------------------------------------------