Transformer-Explainability/BERT_params/cose_bert.json

{
    "max_length": 512,
    "bert_vocab": "bert-base-uncased",
    "bert_dir": "bert-base-uncased",
    "use_evidence_sentence_identifier": 0,
    "use_evidence_token_identifier": 1,
    "evidence_token_identifier": {
        "batch_size": 32,
        "epochs": 10,
        "patience": 10,
        "warmup_steps": 10,
        "lr": 1e-05,
        "max_grad_norm": 0.5,
        "sampling_method": "everything",
        "use_half_precision": 0,
        "cose_data_hack": 1
    },
    "evidence_classifier": {
        "classes": [ "false", "true"],
        "batch_size": 32,
        "warmup_steps": 10,
        "epochs": 10,
        "patience": 10,
        "lr": 1e-05,
        "max_grad_norm": 0.5,
        "sampling_method": "everything",
        "use_half_precision": 0,
        "cose_data_hack": 1
    }
}