Commit
•
65f8cc1
1
Parent(s):
3456945
Update README.md
Browse files
README.md
CHANGED
|
@@ -10,9 +10,11 @@ datasets:
|
|
| 10 |
- samsum
|
| 11 |
---
|
| 12 |
|
| 13 |
-
## bart-base-samsum
|
| 14 |
This model was trained using Amazon SageMaker and the new Hugging Face Deep Learning container.
|
| 15 |
|
|
|
|
|
|
|
| 16 |
For more information look at:
|
| 17 |
- [🤗 Transformers Documentation: Amazon SageMaker](https://huggingface.co/transformers/sagemaker.html)
|
| 18 |
- [Example Notebooks](https://github.com/huggingface/notebooks/tree/master/sagemaker)
|
|
@@ -21,10 +23,38 @@ For more information look at:
|
|
| 21 |
- [Deep Learning Container](https://github.com/aws/deep-learning-containers/blob/master/available_images.md#huggingface-training-containers)
|
| 22 |
|
| 23 |
## Result
|
|
|
|
|
|
|
|
|
|
| 24 |
| key | value |
|
| 25 |
| --- | ----- |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 26 |
| key | value |
|
| 27 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 28 |
|
| 29 |
## Usage
|
| 30 |
```python
|
|
|
|
| 10 |
- samsum
|
| 11 |
---
|
| 12 |
|
| 13 |
+
## `bart-base-samsum`
|
| 14 |
This model was trained using Amazon SageMaker and the new Hugging Face Deep Learning container.
|
| 15 |
|
| 16 |
+
You can find the notebook [here]() and the referring blog post [here]().
|
| 17 |
+
|
| 18 |
For more information look at:
|
| 19 |
- [🤗 Transformers Documentation: Amazon SageMaker](https://huggingface.co/transformers/sagemaker.html)
|
| 20 |
- [Example Notebooks](https://github.com/huggingface/notebooks/tree/master/sagemaker)
|
|
|
|
| 23 |
- [Deep Learning Container](https://github.com/aws/deep-learning-containers/blob/master/available_images.md#huggingface-training-containers)
|
| 24 |
|
| 25 |
## Result
|
| 26 |
+
|
| 27 |
+
### Training
|
| 28 |
+
|
| 29 |
| key | value |
|
| 30 |
| --- | ----- |
|
| 31 |
+
| epoch | 3 |
|
| 32 |
+
| init_mem_cpu_alloc_delta | 180190 |
|
| 33 |
+
| init_mem_cpu_peaked_delta | 18282 |
|
| 34 |
+
| init_mem_gpu_alloc_delta | 558658048 |
|
| 35 |
+
| init_mem_gpu_peaked_delta | 0 |
|
| 36 |
+
| train_mem_cpu_alloc_delta | 6658519 |
|
| 37 |
+
| train_mem_cpu_peaked_delta | 642937 |
|
| 38 |
+
| train_mem_gpu_alloc_delta | 2267624448 |
|
| 39 |
+
| train_mem_gpu_peaked_delta | 10355728896 |
|
| 40 |
+
| train_runtime | 98.4931 |
|
| 41 |
+
| train_samples | 14732 |
|
| 42 |
+
| train_samples_per_second | 3.533 |
|
| 43 |
+
|
| 44 |
+
### Evaluation
|
| 45 |
+
|
| 46 |
| key | value |
|
| 47 |
+
| --- | ----- |
|
| 48 |
+
| epoch | 3 |
|
| 49 |
+
| eval_loss | 1.5356481075286865 |
|
| 50 |
+
| eval_mem_cpu_alloc_delta | 659047 |
|
| 51 |
+
| eval_mem_cpu_peaked_delta | 18254 |
|
| 52 |
+
| eval_mem_gpu_alloc_delta | 0 |
|
| 53 |
+
| eval_mem_gpu_peaked_delta | 300285440 |
|
| 54 |
+
| eval_runtime | 0.3116 |
|
| 55 |
+
| eval_samples | 818 |
|
| 56 |
+
| eval_samples_per_second | 2625.337 |
|
| 57 |
+
|
| 58 |
|
| 59 |
## Usage
|
| 60 |
```python
|