Model save

README.md

@@ -1,7 +1,7 @@
 ---
 library_name: transformers
 license: apache-2.0
-base_model: google/electra-small-discriminator
+base_model: hyt1912/electra-small-discriminator-finetuned-squad
 tags:
 - generated_from_trainer
 model-index:
@@ -14,9 +14,9 @@ should probably proofread and complete it, then remove this comment. -->
 
 # electra-small-discriminator-finetuned-squad
 
-This model is a fine-tuned version of [google/electra-small-discriminator](https://huggingface.co/google/electra-small-discriminator) on an unknown dataset.
+This model is a fine-tuned version of [hyt1912/electra-small-discriminator-finetuned-squad](https://huggingface.co/hyt1912/electra-small-discriminator-finetuned-squad) on an unknown dataset.
 It achieves the following results on the evaluation set:
-- Loss: 1.1425
+- Loss: 1.0458
 
 ## Model description
 
@@ -39,7 +39,7 @@ The following hyperparameters were used during training:
 - train_batch_size: 16
 - eval_batch_size: 16
 - seed: 42
-- optimizer: Use adamw_torch with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
+- 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: 3
 
@@ -47,14 +47,14 @@ The following hyperparameters were used during training:
 
 | Training Loss | Epoch | Step  | Validation Loss |
 |:-------------:|:-----:|:-----:|:---------------:|
-| 1.3266        | 1.0   | 5533  | 1.2489          |
-| 1.1417        | 2.0   | 11066 | 1.1470          |
-| 1.0544        | 3.0   | 16599 | 1.1425          |
+| 1.0184        | 1.0   | 3729  | 1.0553          |
+| 0.9004        | 2.0   | 7458  | 1.0349          |
+| 0.7912        | 3.0   | 11187 | 1.0458          |
 
 
 ### Framework versions
 
-- Transformers 4.46.2
-- Pytorch 2.5.1+cu121
+- Transformers 4.46.3
+- Pytorch 2.5.0
 - Datasets 3.1.0
 - Tokenizers 0.20.3