diff --git a/ppcls/configs/Pedestrian/reid_sb_ResNet50.yaml b/ppcls/configs/Pedestrian/reid_sb_ResNet50.yaml index e8db1bef9914568302701304c87aaae68c4bc880..01f6cac8afebaea2a866e9d4a5e19352b0016ece 100644 --- a/ppcls/configs/Pedestrian/reid_sb_ResNet50.yaml +++ b/ppcls/configs/Pedestrian/reid_sb_ResNet50.yaml @@ -35,8 +35,11 @@ Loss: Train: - CELoss: weight: 1.0 + epsilon: 0.1 - TripletLoss: weight: 1.0 + margin: 0.3 + normalize_feature: false Eval: - CELoss: weight: 1.0