search

XiaShan1227 / Graphormer

Do Transformers Really Perform Bad for Graph Representation? [NIPS-2021]
36 stars 8 forks source link

实例拷贝 #2

Open Aron-04Wang opened 3 weeks ago

Aron-04Wang commented 3 weeks ago

从autodl的镜像找过来，配这个环境已经3天了，看了一下您的代码觉得非常清晰易懂想在自己的电脑上改一改用在自己的项目里，但那里三在base环境里配的不好做yaml复制，请问能不能做一个requirement.txt 或yml文件，谢谢！

XiaShan1227 commented 3 weeks ago

核心库就两个，你用最新的就可以。我这个： torch 2.0.1+cu118 torch_geometric 2.6.1

Aron-04Wang commented 3 weeks ago

python main.py --exp_name=ESOL_Graphormer这个训练一轮要一个小时，处理速度差不多是 295.99s/it，请问这是什么原因

XiaShan1227 commented 2 weeks ago

python main.py --exp_name=ESOL_Graphormer这个训练一轮要一个小时，处理速度差不多是 295.99s/it，请问这是什么原因

代码里面涉及到大量的计算，比如：最短路径（你可以进行图的预处理）；关于边和空间编码，可以考虑改写提高计算效率...