Update README
Browse files
README.md
CHANGED
|
@@ -59,31 +59,12 @@ Jeff: and how can I get started?
|
|
| 59 |
Jeff: where can I find documentation?
|
| 60 |
Philipp: ok, ok you can find everything here. https://huggingface.co/blog/the-partnership-amazon-sagemaker-and-hugging-face
|
| 61 |
'''
|
| 62 |
-
|
| 63 |
```
|
| 64 |
|
| 65 |
## Training procedure
|
| 66 |
- Colab notebook: https://colab.research.google.com/drive/1RInRjLLso9E2HG_xjA6j8JO3zXzSCBRF?usp=sharing
|
| 67 |
|
| 68 |
-
## Hyperparameters
|
| 69 |
-
```json
|
| 70 |
-
{
|
| 71 |
-
"dataset_name": "samsum",
|
| 72 |
-
"do_eval": true,
|
| 73 |
-
"do_predict": true,
|
| 74 |
-
"do_train": true,
|
| 75 |
-
"fp16": true,
|
| 76 |
-
"learning_rate": 2e-05,
|
| 77 |
-
"model_name_or_path": "facebook/bart-base",
|
| 78 |
-
"num_train_epochs": 1,
|
| 79 |
-
"per_device_eval_batch_size": 4,
|
| 80 |
-
"per_device_train_batch_size": 4,
|
| 81 |
-
"predict_with_generate": true,
|
| 82 |
-
"gradient_accumulation_steps": 2,
|
| 83 |
-
"weight_decay": 0.01,
|
| 84 |
-
}
|
| 85 |
-
```
|
| 86 |
-
|
| 87 |
## Results
|
| 88 |
| key | value |
|
| 89 |
| --- | ----- |
|
|
|
|
| 59 |
Jeff: where can I find documentation?
|
| 60 |
Philipp: ok, ok you can find everything here. https://huggingface.co/blog/the-partnership-amazon-sagemaker-and-hugging-face
|
| 61 |
'''
|
| 62 |
+
summarizer(conversation)
|
| 63 |
```
|
| 64 |
|
| 65 |
## Training procedure
|
| 66 |
- Colab notebook: https://colab.research.google.com/drive/1RInRjLLso9E2HG_xjA6j8JO3zXzSCBRF?usp=sharing
|
| 67 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 68 |
## Results
|
| 69 |
| key | value |
|
| 70 |
| --- | ----- |
|