Commit ·
b5ac7fb
1
Parent(s): 966b504
Update README.md
Browse files
README.md
CHANGED
|
@@ -7,4 +7,18 @@ Prompting format:
|
|
| 7 |
[Person 1]: [message]
|
| 8 |
[Person 2]: [message]
|
| 9 |
[Person 1]:
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 10 |
```
|
|
|
|
| 7 |
[Person 1]: [message]
|
| 8 |
[Person 2]: [message]
|
| 9 |
[Person 1]:
|
| 10 |
+
```
|
| 11 |
+
|
| 12 |
+
Training params:
|
| 13 |
+
```py
|
| 14 |
+
transformers.TrainingArguments(
|
| 15 |
+
per_device_train_batch_size=4,
|
| 16 |
+
fp16=True,
|
| 17 |
+
warmup_steps=30,
|
| 18 |
+
max_steps=200,
|
| 19 |
+
learning_rate=1.7e-4,
|
| 20 |
+
logging_steps=1,
|
| 21 |
+
output_dir='outputs',
|
| 22 |
+
report_to="none"
|
| 23 |
+
)
|
| 24 |
```
|