qq_51578371的博客

原创 复现Temporal Knowledge Graph Reasoning with Historical Contrastive Learning

GDELT和WIKI数据集都太大了，本机GPU根本加载数据集会killed,所以下面我用了。在这两个版本上面安装requirements.txt里的包都不会有报错的情况。torch=1.8.1 cuda版本我安装的11.1。目前还在排队中我感觉申请这个GPU是可以跑通了。