model update

README.md

@@ -191,29 +191,28 @@ vector = model.get_embedding(['Tokyo', 'Japan'])  # shape of (1024, )
 ### Training hyperparameters
 
 The following hyperparameters were used during training:
- - model: "roberta-base"
- - max_length: "64"
- - mode: "mask"
- - data: "relbert/semeval2012_relational_similarity_v6"
- - split: "train"
- - split_eval: "validation"
- - template_mode: "manual"
- - template: "Today, I finally discovered the relation between <subj> and <obj> : <mask>"
- - loss_function: "nce_logout"
- - classification_loss: "False"
- - temperature_nce_constant: "0.05"
- - temperature_nce_rank: "{'min': 0.01, 'max': 0.05, 'type': 'linear'}"
- - epoch: "8"
- - batch: "128"
- - lr: "5e-06"
- - lr_decay: "False"
- - lr_warmup: "1"
- - weight_decay: "0"
- - random_seed: "0"
- - exclude_relation: "None"
- - n_sample: "320"
- - gradient_accumulation: "8"
- - relation_level: "None"
+ - model: roberta-base
+ - max_length: 64
+ - mode: mask
+ - data: relbert/semeval2012_relational_similarity_v6
+ - split: train
+ - split_eval: validation
+ - template_mode: manual
+ - loss_function: nce_logout
+ - classification_loss: False
+ - temperature_nce_constant: 0.05
+ - temperature_nce_rank: {'min': 0.01, 'max': 0.05, 'type': 'linear'}
+ - epoch: 8
+ - batch: 128
+ - lr: 5e-06
+ - lr_decay: False
+ - lr_warmup: 1
+ - weight_decay: 0
+ - random_seed: 0
+ - exclude_relation: None
+ - n_sample: 320
+ - gradient_accumulation: 8
+ - relation_level: None
 
 The full configuration can be found at [fine-tuning parameter file](https://huggingface.co/relbert/relbert-roberta-base-semeval2012-v6-mask-prompt-c-nce-0/raw/main/trainer_config.json).