SeyedAli
/

Multilingual-Text-Semantic-Search-Siamese-BERT-V1

Sentence Similarity

sentence-transformers

feature-extraction

text-embeddings-inference

Inference Endpoints

Model card Files Files and versions Community

SeyedAli commited on Sep 26, 2023

Commit

a2bb023

•

1 Parent(s): c1deb70

Update train_script.py

Files changed (1) hide show

train_script.py +1 -1

train_script.py CHANGED Viewed

@@ -291,7 +291,7 @@ class Dataset:
 if __name__ == "__main__":
     parser = argparse.ArgumentParser()
-    parser.add_argument('--model', default='nreimers/MiniLM-L6-H384-uncased')
     parser.add_argument('--steps', type=int, default=2000)
     parser.add_argument('--save_steps', type=int, default=10000)
     parser.add_argument('--batch_size', type=int, default=64)

 if __name__ == "__main__":
     parser = argparse.ArgumentParser()
+    parser.add_argument('--model', default='SeyedAli/Multilingual-Text-Semantic-Search-Siamese-BERT-V1')
     parser.add_argument('--steps', type=int, default=2000)
     parser.add_argument('--save_steps', type=int, default=10000)
     parser.add_argument('--batch_size', type=int, default=64)