Hugging Face's logo

dariolopez
/
roberta-base-bne-finetuned-msmarco-qa-es

Sentence Similarity
sentence-transformers
PyTorch
Spanish
roberta
text-embeddings-inference
Model card Files Community
1
dariolopez commited on
Commit
7c87a7e
·
1 Parent(s): 4e8618d

Update train_config.json

Browse files
Files changed (1) hide show
  1. train_config.json +2 -2
train_config.json CHANGED
@@ -2,13 +2,13 @@
2
  "model_name": "PlanTL-GOB-ES/roberta-base-bne",
3
  "max_seq_length": 512,
4
  "epochs": 10,
5
- "warmup_steps": 1,
6
  "batch_size": 16,
7
  "optimizer_params": {
8
  "lr": 2e-05
9
  },
10
  "loss": "tl",
11
- "dataset_train_size": 512,
12
  "dataset_name": "dariolopez/ms-marco-es-500k",
13
  "seed": 42,
14
  "length_embedding": 768
 
2
  "model_name": "PlanTL-GOB-ES/roberta-base-bne",
3
  "max_seq_length": 512,
4
  "epochs": 10,
5
+ "warmup_steps": 1000,
6
  "batch_size": 16,
7
  "optimizer_params": {
8
  "lr": 2e-05
9
  },
10
  "loss": "tl",
11
+ "dataset_train_size": 500_000,
12
  "dataset_name": "dariolopez/ms-marco-es-500k",
13
  "seed": 42,
14
  "length_embedding": 768