casrel / config.json
woshiyuanshengaoshou's picture
Push model using huggingface_hub.
d5e25cb verified
{
"model_name": "Casrel",
"lr": 1e-05,
"multi_gpu": false,
"dataset": "CMED",
"batch_size": 6,
"max_epoch": 3,
"test_epoch": 5,
"train_prefix": "train_triples",
"dev_prefix": "dev_triples",
"test_prefix": "test_triples",
"max_len": 150,
"rel_num": 44,
"period": 50,
"debug": false,
"model_type": "bert",
"checkpoint_dir": "D:\\kbqa\\Casrel-med\\CasRel-pytorch-reimplement\\CasRel-reimplement\\checkpoint",
"log_dir": "D:\\kbqa\\Casrel-med\\CasRel-pytorch-reimplement\\CasRel-reimplement\\log\\CMED",
"data_path": "D:\\kbqa\\Casrel-med\\CasRel-pytorch-reimplement\\CasRel-reimplement\\data\\CMED",
"save_path": "D:\\kbqa\\Casrel-med\\CasRel-pytorch-reimplement\\CasRel-reimplement\\save\\model_checkpoint\\casrel.pt",
"log_save_name": "log_save"
}