File size: 744 Bytes
603ddce
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
{
  "model_save_dir": "models",
  "model_save_name": "check",
  "opt_model_description": "test",
  "opt_model_lang": "en",
  "train_batch_size": 64,
  "num_epochs": 200,
  "warm_up_perc": 1,
  "learning_rate": 2e-06,
  "loss_type": "supcon",
  "val_perc": 0.2,
  "wandb_names": {
    "project": "linktransformer",
    "id": "your-id",
    "run": "run-name",
    "entity": "your-id"
  },
  "add_pooling_layer": false,
  "large_val": true,
  "eval_steps_perc": 0.1,
  "test_at_end": true,
  "save_val_test_pickles": true,
  "val_query_prop": 0.5,
  "loss_params": {},
  "eval_type": "classification",
  "training_dataset": "dataframe",
  "base_model_path": "sentence-transformers/multi-qa-mpnet-base-dot-v1",
  "best_model_path": "models/check"
}