diff --git a/convlab2/nlg/scgpt/README.md b/convlab2/nlg/scgpt/README.md index 8f67ce56976ef21da9250fe47abe5346ddc8af90..7d32ac05e05d008462439486b37c6d4139e18cc4 100644 --- a/convlab2/nlg/scgpt/README.md +++ b/convlab2/nlg/scgpt/README.md @@ -26,3 +26,17 @@ def generate(self, action) ``` This class supports both CPU and GPU mode by providing the 'device' parameter in constructor function. + + +## Reference +``` +@inproceedings{peng-etal-2020-shot, + title = "Few-shot Natural Language Generation for Task-Oriented Dialog", + author = "Peng, Baolin and Zhu, Chenguang and Li, Chunyuan and Li, Xiujun and Li, Jinchao and Zeng, Michael and Gao, Jianfeng", + booktitle = "Findings of the Association for Computational Linguistics: EMNLP 2020", + month = nov, + year = "2020", + publisher = "Association for Computational Linguistics", + pages = "172--182", +} +``` \ No newline at end of file