Model save

README.md

@@ -1,6 +1,5 @@
 ---
 tags:
-- summarization
 - generated_from_trainer
 model-index:
 - name: BioNLP-intro-disc-eLife
@@ -32,22 +31,18 @@ More information needed
 
 The following hyperparameters were used during training:
 - learning_rate: 1.3739167643078955e-06
-- train_batch_size: 8
-- eval_batch_size: 8
+- train_batch_size: 4
+- eval_batch_size: 4
 - seed: 42
 - distributed_type: multi-GPU
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 - lr_scheduler_type: linear
-- num_epochs: 15
+- num_epochs: 10
 - mixed_precision_training: Native AMP
 
-### Training results
-
-
-
 ### Framework versions
 
-- Transformers 4.35.2
-- Pytorch 1.13.1+cu117
-- Datasets 2.16.1
-- Tokenizers 0.15.2
+- Transformers 4.40.1
+- Pytorch 2.3.0+cu121
+- Datasets 2.18.0
+- Tokenizers 0.19.1

generation_config.json

@@ -4,5 +4,5 @@
   "decoder_start_token_id": 2,
   "eos_token_id": 2,
   "pad_token_id": 1,
-  "transformers_version": "4.35.2"
+  "transformers_version": "4.40.1"
 }