文件名称：BPT:论文“ BP-Transformer的源代码
文件大小：70KB
文件格式：ZIP
更新时间：2024-05-21 11:18:18
Python
BP变压器 此回购包含我们论文的代码 叶子豪，郭启鹏，甘泉，邱锡鹏，张政 该代码是用编写的，并以PyTorch作为后端。 要求 火炬文字0.4 dgl 0.4（master分支上的代码与dgl 0.5不兼容，请签develop dgl 0.5兼容版本的develop分支）。 yaml 虚假 PyTorch 1.1+ 用法 对于多GPU培训，由于提到的PyTorch错误，请在运行脚本之前export NCCL_LL_THRESHOLD=0 。 该代码库具有两个依赖项： graph_kernel和graph_builder ，第一个依赖graph_kernel是用CUDA编写的具有节点并行策略的GPU上的高效图形关注，第二个是用Cython编写的高效图构建。 要安装它们： cd graph_builder python setup.py install cd .. cd graph_