Merge pull request #1217 from chenyuwuxin/master

FIX: code optimization of SRGNN

Author: 2017pxy

recbole/model/sequential_recommender/srgnn.py

@@ -177,7 +177,7 @@ def _get_slice(self, item_seq):
         # The relative coordinates of the item node, shape of [batch_size, max_session_len]
         alias_inputs = torch.LongTensor(alias_inputs).to(self.device)
         # The connecting matrix, shape of [batch_size, max_session_len, 2 * max_session_len]
-        A = torch.FloatTensor(A).to(self.device)
+        A = torch.FloatTensor(np.array(A)).to(self.device)
         # The unique item nodes, shape of [batch_size, max_session_len]
         items = torch.LongTensor(items).to(self.device)