数据集优有一点小问题
hsdjkfnsfc opened this issue · comments
Tang junior commented
Wenyuan commented
@hsdjkfnsfc 谢谢反馈,请问您是从哪个链接下载的数据集和模型,我看一下
Tang junior commented
EmperorKaiser commented
可以写个脚本自行修改一下根目录,我排查了一下,是每个目录下的train.txt,val.txt, test.txt, 还有gt文件夹下的每个pkl里面的image_path字段,这些地方的路径都可以替换为相对路径,或者自己的绝对路径
Wenyuan commented
@EmperorKaiser 谢谢。
@hsdjkfnsfc 抱歉回复比较晚。我在cuda10.2下复现了你的问题,但是切换到cuda9.2后,结果正常。所以,很可能是不同cuda版本下,软件的影响,特别时torch geometric。
我的测试环境:
- cuda 9.2
- torch 1.7.0+torchvision 0.8.0
- torch-cluster 1.5.9
- torch-geometric 1.6.3
- torch-scatter 2.0.6
- torch-sparse 0.6.9
- torch-spline-conv 1.2.1
lmmlzn commented
@hsdjkfnsfc @xuewenyuan @EmperorKaiser 数据集是不是少了一个graph_edge的文件夹,只有gt、graph_node和graph_target文件夹
YongZ-Lee commented
您好,数据集graph_target缺少这个怎么解决呢? @xuewenyuan