fix order

RUCAIBox · Sep 13, 2023 · 11394e9 · 11394e9
1 parent 165b72c
commit 11394e9
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/recbole/model/sequential_recommender/bert4rec.py b/recbole/model/sequential_recommender/bert4rec.py
@@ -134,8 +134,8 @@ def forward(self, item_seq):
             input_emb, extended_attention_mask, output_all_encoded_layers=True
         )
         ffn_output = self.output_ffn(trm_output[-1])
-        ffn_output = self.output_ln(ffn_output)
-        output = self.output_gelu(ffn_output)
+        ffn_output = self.output_gelu(ffn_output)
+        output = self.output_ln(ffn_output)
         return output  # [B L H]
 
     def multi_hot_embed(self, masked_index, max_length):