Training complete

README.md

@@ -19,13 +19,13 @@ should probably proofread and complete it, then remove this comment. -->
 
 This model is a fine-tuned version of [Salesforce/codet5-base](https://huggingface.co/Salesforce/codet5-base) on an unknown dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.3038
-- Rouge1: 0.5548
-- Rouge2: 0.3653
-- Rougel: 0.5284
-- Bleu: 24.1256
-- Meteor: 0.5389
-- Codebleu: 0.2973
+- Loss: 0.2969
+- Rouge1: 0.5412
+- Rouge2: 0.3606
+- Rougel: 0.5202
+- Bleu: 20.5506
+- Meteor: 0.5161
+- Codebleu: 0.2608
 
 ## Model description
 
@@ -45,14 +45,14 @@ More information needed
 
 The following hyperparameters were used during training:
 - learning_rate: 5e-05
-- train_batch_size: 2
-- eval_batch_size: 2
+- train_batch_size: 6
+- eval_batch_size: 12
 - seed: 42
-- gradient_accumulation_steps: 16
-- total_train_batch_size: 32
+- gradient_accumulation_steps: 2
+- total_train_batch_size: 12
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 - lr_scheduler_type: linear
-- num_epochs: 10
+- num_epochs: 2
 - mixed_precision_training: Native AMP
 
 ### Training results

runs/Aug05_05-56-35_cb69df8c9ea7/events.out.tfevents.1722837396.cb69df8c9ea7.4878.0

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:9de1d1c45f638271cf9a9a68cbcc325fb1163c2f77b56b48629e7e2218f7324e
-size 6236
+oid sha256:8e3811ea00d54eac6576e986e813b4400ff347861d5dd623df673645f0caef19
+size 6590

runs/Aug05_05-56-35_cb69df8c9ea7/events.out.tfevents.1722840599.cb69df8c9ea7.4878.1

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7365cc19c3589e5aa447312beae86b4deba72e8622cd872a7e762baa79dcd7bc
+size 659