End of training

README.md

@@ -4,6 +4,8 @@ license: apache-2.0
 base_model: Salesforce/codet5-base
 tags:
 - generated_from_trainer
+metrics:
+- rouge
 model-index:
 - name: CodeT5-KeyPhrases-Filtered-Valid-Phase1
   results: []
@@ -15,6 +17,11 @@ should probably proofread and complete it, then remove this comment. -->
 # CodeT5-KeyPhrases-Filtered-Valid-Phase1
 
 This model is a fine-tuned version of [Salesforce/codet5-base](https://huggingface.co/Salesforce/codet5-base) on the None dataset.
+It achieves the following results on the evaluation set:
+- Loss: 0.9820
+- Rouge1: 0.0972
+- Rouge2: 0.0667
+- Rougel: 0.0972
 
 ## Model description
 
@@ -39,11 +46,18 @@ The following hyperparameters were used during training:
 - seed: 42
 - optimizer: Use OptimizerNames.ADAMW_TORCH with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
 - lr_scheduler_type: linear
-- num_epochs: 4
+- num_epochs: 5
 - mixed_precision_training: Native AMP
 
 ### Training results
 
+| Training Loss | Epoch | Step | Validation Loss | Rouge1 | Rouge2 | Rougel |
+|:-------------:|:-----:|:----:|:---------------:|:------:|:------:|:------:|
+| 5.0107        | 1.0   | 1    | 6.5380          | 0.1056 | 0.0    | 0.1056 |
+| 5.0799        | 2.0   | 2    | 6.5380          | 0.1056 | 0.0    | 0.1056 |
+| 5.2912        | 3.0   | 3    | 1.1345          | 0.1472 | 0.0667 | 0.1472 |
+| 2.3361        | 4.0   | 4    | 1.0165          | 0.0972 | 0.0667 | 0.0972 |
+| 1.5034        | 5.0   | 5    | 0.9820          | 0.0972 | 0.0667 | 0.0972 |
 
 
 ### Framework versions

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:1e34f6b93f8898014a7591d0b6973aed64b9094cc17ffd307d26847c3e5e38a3
+oid sha256:f30bd7514a89ece80588fd76f09ab4fb9be1b5fb811c3455e4ae86c2fbac966d
 size 891558696

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:0ef5c7c2a84deb82375ac9ba8758e17194c133b4c7e75318129f6a20522bce46
+oid sha256:315732a2f47b97bd3c5906c0771d0780b3ec5084c7eb263103626bd668e212ee
 size 5496