TypeError: ‘weights_only‘ is an invalid keyword argument for Unpickler()解决
使用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.")