joe32140 commited on
Commit
9b611fd
1 Parent(s): b610ec2

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -0
README.md CHANGED
@@ -21,6 +21,8 @@ tags:
21
 
22
  This is a [PyLate](https://github.com/lightonai/pylate) model finetuned from [answerdotai/ModernBERT-base](https://huggingface.co/answerdotai/ModernBERT-base) on the [train](https://huggingface.co/datasets/lightonai/ms-marco-en-bge) dataset. It maps sentences & paragraphs to sequences of 128-dimensional dense vectors and can be used for semantic textual similarity using the MaxSim operator.
23
 
 
 
24
  ## Model Details
25
 
26
  ### Model Description
 
21
 
22
  This is a [PyLate](https://github.com/lightonai/pylate) model finetuned from [answerdotai/ModernBERT-base](https://huggingface.co/answerdotai/ModernBERT-base) on the [train](https://huggingface.co/datasets/lightonai/ms-marco-en-bge) dataset. It maps sentences & paragraphs to sequences of 128-dimensional dense vectors and can be used for semantic textual similarity using the MaxSim operator.
23
 
24
+ I finetuned the model with official script [examples/train_pylate.py](https://github.com/AnswerDotAI/ModernBERT/blob/main/examples/train_pylate.py) on a RTX 4090 GPU in 12 hours. See more details in [trianing logs](https://wandb.ai/joe32140/col-modernbert-msmarco/workspace?nw=nwuserjoe32140).
25
+
26
  ## Model Details
27
 
28
  ### Model Description