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 |
| --- | ----- |
|