diff --git a/models/rank/dien/model.py b/models/rank/dien/model.py index 0a2d1c7f6b18158492ed5c739203da39c5181f49..74dc97fc4b648fa1d4545520c1033174e44b797e 100755 --- a/models/rank/dien/model.py +++ b/models/rank/dien/model.py @@ -230,7 +230,7 @@ class Model(ModelBase): pad_value) seq_shape = fluid.layers.shape(pos_seq_pad) - if(seq_shape[1] < 2): + if seq_shape[1] < 2: aux_loss = 0 else: test_pos = fluid.layers.reduce_sum(