Update train_script.py
Browse files- train_script.py +1 -1
train_script.py
CHANGED
|
@@ -291,7 +291,7 @@ class Dataset:
|
|
| 291 |
|
| 292 |
if __name__ == "__main__":
|
| 293 |
parser = argparse.ArgumentParser()
|
| 294 |
-
parser.add_argument('--model', default='
|
| 295 |
parser.add_argument('--steps', type=int, default=2000)
|
| 296 |
parser.add_argument('--save_steps', type=int, default=10000)
|
| 297 |
parser.add_argument('--batch_size', type=int, default=64)
|
|
|
|
| 291 |
|
| 292 |
if __name__ == "__main__":
|
| 293 |
parser = argparse.ArgumentParser()
|
| 294 |
+
parser.add_argument('--model', default='SeyedAli/Multilingual-Text-Semantic-Search-Siamese-BERT-V1')
|
| 295 |
parser.add_argument('--steps', type=int, default=2000)
|
| 296 |
parser.add_argument('--save_steps', type=int, default=10000)
|
| 297 |
parser.add_argument('--batch_size', type=int, default=64)
|