training_args.json

{
  "model_name": "/home/pablo/affilgood/affilgood/entity_linking/data/contrastive/model/finetuned_with_iter0_100percROR_special_tokens",
  "data_file": "/home/pablo/affilgood/affilgood/entity_linking/data/contrastive/dataset/iter1_thres60_5neg_more-labels_100percROR_special_tokens.json",
  "output_dir": "/home/pablo/affilgood/affilgood/entity_linking/data/contrastive/model/finetuned_with_iter1_thres60_5neg_more-labels_100percROR_special_tokens",
  "batch_size": 16,
  "epochs": 3,
  "learning_rate": 2e-05,
  "warmup_steps": 1000,
  "max_seq_length": 128,
  "eval_steps": 1000,
  "seed": 42,
  "use_amp": true,
  "num_hard_negatives": 5,
  "temperature": 0.05
}