MTL KGC:编码“具有预训练语言模型的知识图完成多任务学习”的代码 源码

货柜码头 这是的PyTorch实施。 火车 使用链接预测(LP),关系预测(RP)和相关性排名(RR)训练多任务学习。 如果收到AssertionError:未初始化默认进程组,请尝试使用python -m torch.distributed.launch python run_bert_multitask.py \ --do_train \ --task_list lp,rp,rr \ --data_dir ./data/wn18rr \ --bert_model bert-base-cased \ --max_seq_length 128 \ --output_dir ./output_dir \ --num_train_epochs 5.0 \ --learning_rate=2e-5 \ --tb_log_