diff --git a/recbole/model/context_aware_recommender/autoint.py b/recbole/model/context_aware_recommender/autoint.py index 893d5d8f2..177a159b3 100644 --- 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()