Commit
•
38a5b89
1
Parent(s):
5918ef4
Update README.md
Browse files
README.md
CHANGED
|
@@ -30,9 +30,24 @@ For more information look at:
|
|
| 30 |
- [Python SDK SageMaker documentation for Hugging Face](https://sagemaker.readthedocs.io/en/stable/frameworks/huggingface/index.html)
|
| 31 |
- [Deep Learning Container](https://github.com/aws/deep-learning-containers/blob/master/available_images.md#huggingface-training-containers)
|
| 32 |
|
| 33 |
-
##
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 34 |
|
| 35 |
-
|
| 36 |
|
| 37 |
| key | value |
|
| 38 |
| --- | ----- |
|
|
@@ -49,7 +64,7 @@ For more information look at:
|
|
| 49 |
| train_samples | 14732 |
|
| 50 |
| train_samples_per_second | 3.533 |
|
| 51 |
|
| 52 |
-
|
| 53 |
|
| 54 |
| key | value |
|
| 55 |
| --- | ----- |
|
| 30 |
- [Python SDK SageMaker documentation for Hugging Face](https://sagemaker.readthedocs.io/en/stable/frameworks/huggingface/index.html)
|
| 31 |
- [Deep Learning Container](https://github.com/aws/deep-learning-containers/blob/master/available_images.md#huggingface-training-containers)
|
| 32 |
|
| 33 |
+
## Hyperparameters
|
| 34 |
+
```json
|
| 35 |
+
{
|
| 36 |
+
"dataset_name": "samsum",
|
| 37 |
+
"do_eval": true,
|
| 38 |
+
"do_train": true,
|
| 39 |
+
"fp16": true,
|
| 40 |
+
"learning_rate": 5e-05,
|
| 41 |
+
"model_name_or_path": "facebook/bart-base",
|
| 42 |
+
"num_train_epochs": 3,
|
| 43 |
+
"output_dir": "/opt/ml/model",
|
| 44 |
+
"per_device_eval_batch_size": 8,
|
| 45 |
+
"per_device_train_batch_size": 8,
|
| 46 |
+
"seed": 7
|
| 47 |
+
}
|
| 48 |
+
```
|
| 49 |
|
| 50 |
+
## Train results
|
| 51 |
|
| 52 |
| key | value |
|
| 53 |
| --- | ----- |
|
| 64 |
| train_samples | 14732 |
|
| 65 |
| train_samples_per_second | 3.533 |
|
| 66 |
|
| 67 |
+
## Eval results
|
| 68 |
|
| 69 |
| key | value |
|
| 70 |
| --- | ----- |
|