36 lines
1.3 KiB
JSON
36 lines
1.3 KiB
JSON
|
{
|
||
|
|
// Use IntelliSense to learn about possible attributes.
|
||
|
|
// Hover to view descriptions of existing attributes.
|
||
|
|
// For more information, visit: https://go.microsoft.com/fwlink/?linkid=830387
|
||
|
|
"version": "0.2.0",
|
||
|
|
"configurations": [
|
||
|
|
{
|
||
|
|
"name": "Python: Current File",
|
||
|
|
"type": "python",
|
||
|
|
"request": "launch",
|
||
|
|
"program": "${file}",
|
||
|
|
"console": "integratedTerminal",
|
||
|
|
"justMyCode": true,
|
||
|
|
"args": [
|
||
|
|
"--gpus", "1,",
|
||
|
|
"--max_epochs=16",
|
||
|
|
"--num_workers=32",
|
||
|
|
"--model_name_or_path", "bert-base-uncased",
|
||
|
|
"--accumulate_grad_batches", "1",
|
||
|
|
"--model_class", "BertKGC",
|
||
|
|
"--batch_size", "32",
|
||
|
|
"--checkpoint", "/root/kg_374/Relphormer/pretrain/output/FB15k-237/epoch=15-step=38899-Eval/hits10=0.96.ckpt",
|
||
|
|
"--pretrain", "0",
|
||
|
|
"--bce", "0",
|
||
|
|
"--check_val_every_n_epoch", "1",
|
||
|
|
"--data_dir", "dataset/FB15k-237",
|
||
|
|
"--eval_batch_size", "128",
|
||
|
|
"--max_seq_length", "128",
|
||
|
|
"--lr", "3e-5",
|
||
|
|
"--max_triplet", "64",
|
||
|
|
"--add_attn_bias", "True",
|
||
|
|
"--use_global_node", "True",
|
||
|
|
]
|
||
|
|
}
|
||
|
|
]
|
||
|
|
}
|