trainer_config.json

{"model": "roberta-large", "max_length": 64, "mode": "average", "data": "relbert/semeval2012_relational_similarity", "split": "train", "data_eval": "relbert/conceptnet_high_confidence", "split_eval": "full", "template_mode": "manual", "template": "I wasn\u2019t aware of this relationship, but I just read in the encyclopedia that <subj> is the <mask> of <obj>", "loss_function": "nce_logout", "classification_loss": false, "temperature_nce_constant": 0.05, "temperature_nce_rank": {"min": 0.01, "max": 0.05, "type": "linear"}, "epoch": 26, "batch": 128, "lr": 5e-06, "lr_decay": false, "lr_warmup": 1, "weight_decay": 0, "random_seed": 0, "exclude_relation": null, "exclude_relation_eval": null, "n_sample": 640, "gradient_accumulation": 8}