File size: 666 Bytes
8cb3bc8 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 | {
"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
} |