dariolopez commited on
Commit
ddd9714
1 Parent(s): 039e06a

Upload train_config.json

Browse files
Files changed (1) hide show
  1. train_config.json +15 -0
train_config.json ADDED
@@ -0,0 +1,15 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
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
15
+ }