From f0494806a648693aa22daf6a2443f5ce11c81a40 Mon Sep 17 00:00:00 2001
From: Bowen Zheng <57226275+zhengbw0324@users.noreply.github.com>
Date: Wed, 28 Aug 2024 16:38:46 +0800
Subject: [PATCH] Update simplex.py

---
 recbole/model/general_recommender/simplex.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/recbole/model/general_recommender/simplex.py b/recbole/model/general_recommender/simplex.py
index 9fa9e12c3..53a7929f7 100644
--- 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()