From 390c305bb4c5d1969ef642a1b13f2910b7806e91 Mon Sep 17 00:00:00 2001 From: Sherry-XLL Date: Mon, 4 Apr 2022 02:58:12 +0000 Subject: [PATCH] FIX: fix UserWarning in get_norm_adj_mat --- recbole/model/general_recommender/ncl.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/recbole/model/general_recommender/ncl.py b/recbole/model/general_recommender/ncl.py index 5006b4c7f..1c78c5fcd 100644 --- a/recbole/model/general_recommender/ncl.py +++ b/recbole/model/general_recommender/ncl.py @@ -121,7 +121,7 @@ def get_norm_adj_mat(self): L = sp.coo_matrix(L) row = L.row col = L.col - i = torch.LongTensor([row, col]) + i = torch.LongTensor(np.array([row, col])) data = torch.FloatTensor(L.data) SparseL = torch.sparse.FloatTensor(i, data, torch.Size(L.shape)) return SparseL