angelitasr
/

bge-base-en-v1.5_eeid

Sentence Similarity

sentence-transformers

feature-extraction

Generated from Trainer

dataset_size:3503

loss:MultipleNegativesRankingLoss

text-embeddings-inference

Inference Endpoints

Model card Files Files and versions Community

angelitasr commited on Dec 7, 2024

Commit

df6d12e

·

verified ·

1 Parent(s): 5c2cf9a

End of training

Files changed (1) hide show

README.md +10 -6

README.md CHANGED Viewed

@@ -236,7 +236,7 @@ You can finetune this model on your own dataset.
 ### Training Hyperparameters
 #### Non-Default Hyperparameters
-- `num_train_epochs`: 5
 - `fp16`: True
 - `push_to_hub`: True
 - `batch_sampler`: no_duplicates
@@ -261,7 +261,7 @@ You can finetune this model on your own dataset.
 - `adam_beta2`: 0.999
 - `adam_epsilon`: 1e-08
 - `max_grad_norm`: 1.0
-- `num_train_epochs`: 5
 - `max_steps`: -1
 - `lr_scheduler_type`: linear
 - `lr_scheduler_kwargs`: {}
@@ -362,10 +362,14 @@ You can finetune this model on your own dataset.
 ### Training Logs
 | Epoch  | Step | Training Loss |
 |:------:|:----:|:-------------:|
-| 1.1416 | 500  | 0.3382        |
-| 2.2831 | 1000 | 0.1004        |
-| 3.4247 | 1500 | 0.0386        |
-| 4.5662 | 2000 | 0.0133        |
 ### Framework Versions

 ### Training Hyperparameters
 #### Non-Default Hyperparameters
+- `num_train_epochs`: 10
 - `fp16`: True
 - `push_to_hub`: True
 - `batch_sampler`: no_duplicates
 - `adam_beta2`: 0.999
 - `adam_epsilon`: 1e-08
 - `max_grad_norm`: 1.0
+- `num_train_epochs`: 10
 - `max_steps`: -1
 - `lr_scheduler_type`: linear
 - `lr_scheduler_kwargs`: {}
 ### Training Logs
 | Epoch  | Step | Training Loss |
 |:------:|:----:|:-------------:|
+| 1.1416 | 500  | 0.341         |
+| 2.2831 | 1000 | 0.1082        |
+| 3.4247 | 1500 | 0.0485        |
+| 4.5662 | 2000 | 0.0226        |
+| 5.7078 | 2500 | 0.0133        |
+| 6.8493 | 3000 | 0.0066        |
+| 7.9909 | 3500 | 0.0042        |
+| 9.1324 | 4000 | 0.0017        |
 ### Framework Versions