Fix layer definition bug.

RUCAIBox · Aug 25, 2022 · 07d80fe · 07d80fe
1 parent 61848c0
commit 07d80fe
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/recbole/model/context_aware_recommender/autoint.py b/recbole/model/context_aware_recommender/autoint.py
@@ -52,7 +52,7 @@ def __init__(self, config, dataset):
         self.attn_fc = torch.nn.Linear(self.atten_output_dim, 1)
         self.deep_predict_layer = nn.Linear(self.mlp_hidden_size[-1], 1)
         if self.has_residual:
-            self.v_res_res_embedding = torch.nn.Linear(self.embedding_size, self.attention_size)
+            self.v_res_embedding = torch.nn.Linear(self.embedding_size, self.attention_size)
 
         self.dropout_layer = nn.Dropout(p=self.dropout_probs[2])
         self.sigmoid = nn.Sigmoid()