From 49094dbdde6a13928acd3acea938e57bb5e934a0 Mon Sep 17 00:00:00 2001 From: Alex Barghi Date: Tue, 12 Dec 2023 16:57:28 +0000 Subject: [PATCH] x --- benchmarks/cugraph-pyg/datasets/ogbn_papers100M.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/benchmarks/cugraph-pyg/datasets/ogbn_papers100M.py b/benchmarks/cugraph-pyg/datasets/ogbn_papers100M.py index 0a7c7fa8cb7..5c1035799b2 100644 --- a/benchmarks/cugraph-pyg/datasets/ogbn_papers100M.py +++ b/benchmarks/cugraph-pyg/datasets/ogbn_papers100M.py @@ -108,7 +108,7 @@ def x_dict(self) -> Dict[str, torch.Tensor]: full_path = os.path.join(node_type_path, 'node_feat.npy') else: full_path = os.path.join(node_type_path, f'node_feat_{self.__replication_factor}x.npy') - + self.__disk_x = {'paper': np.load( full_path, mmap_mode='r'