Update model.py

models/rank/dien/model.py

@@ -238,25 +238,28 @@ class Model(ModelBase):
                     fluid.layers.log(
                         fluid.layers.sigmoid(
                             fluid.layers.reduce_sum(
-                            gru_out_pad[:, start_value:seq_shape[1] - 1, :] *
-                            pos_seq_pad[:, start_value + 1:seq_shape[1], :],
+                                gru_out_pad[:, start_value:seq_shape[1] - 1, :]
+                                * pos_seq_pad[:, start_value + 1:seq_shape[
+                                    1], :],
                                 dim=2,
                                 keep_dim=True))),
                     dim=2),
                 dim=1,
                 keep_dim=True)
+            
             test_neg = fluid.layers.reduce_sum(
-            fluid.layers.reduce_sum(
                 fluid.layers.log(
                     fluid.layers.sigmoid(
                         fluid.layers.reduce_sum(
-                            gru_out_pad[:, start_value:seq_shape[1] - 1, :] *
-                            neg_seq_pad[:, start_value + 1:seq_shape[1], :],
+                            gru_out_pad[:, start_value:seq_shape[1] - 1, :]
+                            * neg_seq_pad[:, start_value + 1:seq_shape[
+                                1], :],
                             dim=2,
                             keep_dim=True))),
                 dim=2),
             dim=1,
             keep_dim=True)
+            
             aux_loss = fluid.layers.mean(test_neg + test_pos)
 
         # ------------------------- Interest Evolving Layer (GRU with attentional input (AIGRU)) --------------------------