update model card README.md
Browse files
README.md
CHANGED
|
@@ -34,17 +34,24 @@ More information needed
|
|
| 34 |
|
| 35 |
The following hyperparameters were used during training:
|
| 36 |
- learning_rate: 2e-05
|
| 37 |
-
- train_batch_size:
|
| 38 |
-
- eval_batch_size:
|
| 39 |
- seed: 42
|
| 40 |
- optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
|
| 41 |
- lr_scheduler_type: linear
|
| 42 |
- num_epochs: 1
|
| 43 |
- mixed_precision_training: Native AMP
|
| 44 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 45 |
### Framework versions
|
| 46 |
|
| 47 |
- Transformers 4.13.0.dev0
|
| 48 |
-
- Pytorch 1.
|
| 49 |
-
- Datasets 1.
|
| 50 |
- Tokenizers 0.10.3
|
|
|
|
| 34 |
|
| 35 |
The following hyperparameters were used during training:
|
| 36 |
- learning_rate: 2e-05
|
| 37 |
+
- train_batch_size: 4
|
| 38 |
+
- eval_batch_size: 4
|
| 39 |
- seed: 42
|
| 40 |
- optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
|
| 41 |
- lr_scheduler_type: linear
|
| 42 |
- num_epochs: 1
|
| 43 |
- mixed_precision_training: Native AMP
|
| 44 |
|
| 45 |
+
### Training results
|
| 46 |
+
|
| 47 |
+
| Training Loss | Epoch | Step | Validation Loss | Rouge1 | Rouge2 | Rougel | Rougelsum | Gen Len |
|
| 48 |
+
|:-------------:|:-----:|:----:|:---------------:|:------:|:------:|:-------:|:---------:|:-------:|
|
| 49 |
+
| No log | 1.0 | 250 | 2.9192 | 19.063 | 3.3814 | 14.6509 | 14.7209 | 18.765 |
|
| 50 |
+
|
| 51 |
+
|
| 52 |
### Framework versions
|
| 53 |
|
| 54 |
- Transformers 4.13.0.dev0
|
| 55 |
+
- Pytorch 1.10.0+cu111
|
| 56 |
+
- Datasets 1.16.2.dev0
|
| 57 |
- Tokenizers 0.10.3
|