使用torch_geometric.datasets的Planetoid,提前下载了数据放在raw目录,第一次执行没问题,第二次执行就报TypeError: 'weights_only' is an invalid keyword argument for Unpickler()。
定位了下发现是torch_geometric.datasets的Dataset有问题,其_process函数中的torch.load(f, weights_only=False)用法有问题,删掉weights_only=False即可,更改后代码如下:
def _process(self):
f = osp.join(self.processed_dir, 'pre_transform.pt')
if osp.exists(f) and torch.load(f) != _repr(
self.pre_transform):
warnings.warn(
"The `pre_transform` argument differs from the one used in "
"the pre-processed version of this dataset. If you want to "
"make use of another pre-processing technique, pass "
"`force_reload=True` explicitly to reload the dataset.")
f = osp.join(self.processed_dir, 'pre_filter.pt')
if osp.exists(f) and torch.load(f) != _repr(
self.pre_filter):
warnings.warn(
"The `pre_filter` argument differs from the one used in "
"the pre-processed version of this dataset. If you want to "
"make use of another pre-fitering technique, pass "
"`force_reload=True` explicitly to reload the dataset.")