Commit
·
d0aa4e4
1
Parent(s):
31824a0
Create README.md
Browse files
README.md
ADDED
|
@@ -0,0 +1,8 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
Base model: [microsoft/DialoGPT-small](https://huggingface.co/microsoft/DialoGPT-small)
|
| 2 |
+
|
| 3 |
+
Fine tuned for dialogue response generation on the [Schema Guided Dialogue Dataset](https://github.com/google-research-datasets/dstc8-schema-guided-dialogue) (Rastogi et al., 2019)
|
| 4 |
+
|
| 5 |
+
Three additional special tokens were added during the fine-tuning process:
|
| 6 |
+
- <|pad|> padding token
|
| 7 |
+
- <|user|> speaker control token to prompt user responses
|
| 8 |
+
- <|system|> speaker control token to prompt system responses
|