Update simplex.py

RUCAIBox · Aug 28, 2024 · f049480 · f049480
1 parent 14430b3
commit f049480
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()