| save_dir: save_results |
|
|
| data_dir: data/OIE2016(processed)/relation_model/ |
| train_file: train.json |
| dev_file: dev.json |
| test_file: test.json |
| ent_rel_file: data/ent_rel_file.json |
| max_sent_len: 512 |
| max_wordpiece_len: 512 |
|
|
| embedding_model: bert |
| mlp_hidden_size: 150 |
| max_span_length: 10 |
| dropout: 0.4 |
| logit_dropout: 0.2 |
| bert_model_name: bert-base-uncased |
| bert_output_size: 0 |
| bert_dropout: 0.0 |
|
|
| gradient_clipping: 5.0 |
| learning_rate: 5e-5 |
| bert_learning_rate: 5e-5 |
| lr_decay_rate: 0.9 |
| adam_beta1: 0.9 |
| adam_beta2: 0.9 |
| adam_epsilon: 1e-12 |
| adam_weight_decay_rate: 1e-5 |
| adam_bert_weight_decay_rate: 1e-5 |
|
|
| seed: 5216 |
| epochs: 20 |
| pretrain_epochs: 5 |
| warmup_rate: 0.2 |
| early_stop: 30 |
| train_batch_size: 32 |
| test_batch_size: 32 |
| gradient_accumulation_steps: 1 |
| logging_steps: 32 |
| validate_every: 4032 |
| device: -1 |
| log_file: linking_model_train.log |
|
|
