Update README.md
Browse files
README.md
CHANGED
@@ -48,12 +48,17 @@ More information needed
|
|
48 |
|
49 |
## Training and evaluation data
|
50 |
|
51 |
-
|
|
|
52 |
|
53 |
## Training procedure
|
54 |
|
|
|
|
|
55 |
### Training hyperparameters
|
56 |
|
|
|
|
|
57 |
The following hyperparameters were used during training:
|
58 |
- learning_rate: 5.6e-05
|
59 |
- train_batch_size: 8
|
|
|
48 |
|
49 |
## Training and evaluation data
|
50 |
|
51 |
+
The training data were 10000 samples from the multi-news training dataset
|
52 |
+
and the evaluation data were 500 samples from the multi-news evaluation dataset
|
53 |
|
54 |
## Training procedure
|
55 |
|
56 |
+
For the training procedure the Seq2SeqTrainer class was used from the transformers library.
|
57 |
+
|
58 |
### Training hyperparameters
|
59 |
|
60 |
+
The Hyperparameters were passed to the Seq2SeqTrainingArguments class from the transformers library.
|
61 |
+
|
62 |
The following hyperparameters were used during training:
|
63 |
- learning_rate: 5.6e-05
|
64 |
- train_batch_size: 8
|