dbg

ppcls/engine/trainer.py

@@ -353,12 +353,12 @@ class Trainer(object):
                 block_fea, gallery_feas, transpose_y=True)
             if query_camera_id is not None:
                 camera_id_block = camera_id_blocks[block_idx]
-                camera_id_same = (camera_id_block == gallery_camera_id.t())
+                camera_id_same = (camera_id_block != gallery_camera_id.t())
 
                 image_id_block = image_id_blocks[block_idx]
-                image_id_same = (image_id_block == gallery_img_id.t())
+                image_id_same = (image_id_block != gallery_img_id.t())
 
-                keep_mask = not (camera_id_same and image_id_same)
+                keep_mask = paddle.logical_or(camera_id_same, image_id_same)
                 similarity_matrix = similarity_matrix * keep_mask.astype(
                     "float32")
             if cum_similarity_matrix is None: