fix typos

Author: yule-BUAA

models/MemoryModel.py

@@ -553,7 +553,7 @@ def __init__(self, node_raw_features: torch.Tensor, edge_raw_features: torch.Ten
         """
         Graph attention embedding module.
         :param node_raw_features: Tensor, shape (num_nodes + 1, node_feat_dim)
-        :param edge_raw_features: Tensor, shape (num_edges + 1, node_feat_dim)
+        :param edge_raw_features: Tensor, shape (num_edges + 1, edge_feat_dim)
         :param neighbor_sampler: NeighborSampler, neighbor sampler
         :param time_encoder: TimeEncoder
         :param node_feat_dim: int, dimension of node features