出现该问题是由于torch 1.2版本出了问题。
首先查看自己torch 版本,若为1.2则更新torch
参考链接:
https://github.com/amiryanj/StanfordDroneDataset
import torch
print(torch.__version__)
#输出结果为1.2,则更新
#1.2.0
pip install --upgrade torch-cluster
出现该问题是由于torch 1.2版本出了问题。
首先查看自己torch 版本,若为1.2则更新torch
参考链接:
https://github.com/amiryanj/StanfordDroneDataset
import torch
print(torch.__version__)
#输出结果为1.2,则更新
#1.2.0
pip install --upgrade torch-cluster