Merge pull request #2079 from zhengbw0324/master

Update simplex.py
RUCAIBox · Aug 28, 2024 · d64724a · d64724a
2 parents 29bbe4c + f049480
commit d64724a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/recbole/model/general_recommender/simplex.py b/recbole/model/general_recommender/simplex.py
@@ -74,7 +74,7 @@ def __init__(self, config, dataset):
             if self.aggregator == "self_attention":
                 self.W_q = nn.Linear(self.embedding_size, 1, bias=False)
         # dropout
-        self.dropout_prob = nn.Dropout(config["dropout_prob"])
+        self.dropout = nn.Dropout(config["dropout_prob"])
         self.require_pow = config["require_pow"]
         # l2 regularization loss
         self.reg_loss = EmbLoss()