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 9fa9e12c..53a7929f 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()