Datawhale 图神经网络 Task07 图预测任务实践

非常感谢“天国之影”共享的结果文件，让我们得以看到最终的结果。
我尝试了很久之后，已经放弃跑图预测任务实践的代码了。
首先用tensorboard预览最终的结果：

tensorboard --logdir=GINGraphPooling

可以从tensorboard里看到训练了五个多小时，最终曲线比较平稳。
我们来从代码看看本次实践的任务：

首先是run.sh，linux里bash命令即可，windows的话没有也没关系，因为run.sh里只是运行了python main及task_name,device,num_layers等调参。直接使用python解释器运行main.py也是可以的。


main.py是程序的主体，主要分为接收参数和运行

if __name__ == "__main__":
    args = parse_args()
    main(args)
 

但是这里有个坑，我是再centos上跑的，到现在也没有想清楚为什么？
main函数中ogb.lsc的这个包导入要放在前面，否则报错

硬件要求太高，我这儿直接kill进程了

听说colab也跑不了，要把虚拟内存扩大128G才阔以。

代码结构图