Training complete

README.md

@@ -3,6 +3,7 @@ library_name: transformers
 license: gpl-3.0
 base_model: ckiplab/gpt2-tiny-chinese
 tags:
+- summarization
 - generated_from_trainer
 model-index:
 - name: ADL_HW2_GPT2
@@ -18,6 +19,8 @@ should probably proofread and complete it, then remove this comment. -->
 # ADL_HW2_GPT2
 
 This model is a fine-tuned version of [ckiplab/gpt2-tiny-chinese](https://huggingface.co/ckiplab/gpt2-tiny-chinese) on an unknown dataset.
+It achieves the following results on the evaluation set:
+- Loss: 3.4578
 
 ## Model description
 
@@ -44,6 +47,20 @@ The following hyperparameters were used during training:
 - lr_scheduler_type: linear
 - num_epochs: 8
 
+### Training results
+
+| Training Loss | Epoch | Step | Validation Loss |
+|:-------------:|:-----:|:----:|:---------------:|
+| 4.02          | 1.0   | 543  | 3.6837          |
+| 3.8249        | 2.0   | 1086 | 3.5804          |
+| 3.7431        | 3.0   | 1629 | 3.5336          |
+| 3.6958        | 4.0   | 2172 | 3.5036          |
+| 3.6633        | 5.0   | 2715 | 3.4815          |
+| 3.639         | 6.0   | 3258 | 3.4722          |
+| 3.622         | 7.0   | 3801 | 3.4597          |
+| 3.6095        | 8.0   | 4344 | 3.4578          |
+
+
 ### Framework versions
 
 - Transformers 4.44.2