19 lines
588 B
Bash
Executable File
19 lines
588 B
Bash
Executable File
nohup python -u main.py --gpus "2," --max_epochs=16 --num_workers=32 \
|
|
--model_name_or_path bert-base-uncased \
|
|
--accumulate_grad_batches 1 \
|
|
--model_class BertKGC \
|
|
--batch_size 64 \
|
|
--checkpoint /kg_374/Relphormer/output/FB15k-237/epoch=1-Eval/hits10=Eval/hits1=0.47-Eval/hits1=0.22.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 \
|
|
>logs/train_fb15k-237.log 2>&1 &
|
|
|